MCPcopy Create free account
hub / github.com/tensorflow/tfjs / setupFunc

Function setupFunc

tfjs-backend-wasm/src/kernels/binary_kernel.ts:32–43  ·  view source on GitHub ↗
(backend: BackendWasm)

Source from the content-addressed store, hash-verified

30 void;
31
32 function setupFunc(backend: BackendWasm): void {
33 wasmFunc = backend.wasm.cwrap(kernelName, null /* void */, [
34 'number', // a_id,
35 'array', // a_shape
36 'number', // a_shape.length
37 'number', // b_id
38 'array', // b_shape
39 'number', // b_shape.length
40 'number', // dtype
41 'number' // out_id
42 ]);
43 }
44
45 function kernelFunc(args: {backend: BackendWasm, inputs: BinaryInputs}):
46 TensorInfo {

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…