Function kernelFunc

tfjs-backend-wasm/src/kernels/unary_kernel.ts:36–50

(args: {backend: BackendWasm, inputs: UnaryInputs})

  }

  function kernelFunc(args: {backend: BackendWasm, inputs: UnaryInputs}):
      TensorInfo {
    const {backend, inputs: {x}} = args;
    const xId = backend.dataIdMap.get(x.dataId).id;
    const out = backend.makeOutput(x.shape, outType || x.dtype);
    const outId = backend.dataIdMap.get(out.dataId).id;

    // Short-circuit zero-sized tensors.
    if (util.sizeFromShape(out.shape) === 0) {
      return out;
    }

    wasmFunc(xId, CppDType[x.dtype], outId);
    return out;
  }

  return {kernelName, backendName: 'wasm', setupFunc, kernelFunc};
}