MCPcopy
hub / github.com/tensorflow/tfjs / compileAndRun

Method compileAndRun

tfjs-backend-webgl/src/backend_webgl.ts:1017–1026  ·  view source on GitHub ↗
(
      program: GPGPUProgram, inputs: TensorInfo[], outputDtype?: DataType,
      customUniformValues?: number[][],
      preventEagerUnpackingOfOutput = false)

Source from the content-addressed store, hash-verified

1015 }
1016
1017 compileAndRun(
1018 program: GPGPUProgram, inputs: TensorInfo[], outputDtype?: DataType,
1019 customUniformValues?: number[][],
1020 preventEagerUnpackingOfOutput = false): TensorInfo {
1021 outputDtype = outputDtype || inputs[0].dtype;
1022 const outInfo = this.runWebGLProgram(
1023 program, inputs, outputDtype, customUniformValues,
1024 preventEagerUnpackingOfOutput);
1025 return outInfo;
1026 }
1027
1028 private getAndSaveBinary(key: string, getBinary: () => GPGPUBinary):
1029 GPGPUBinary {

Callers 5

packedUnaryOpMethod · 0.95
absMethod · 0.95
squareAndAddFunction · 0.80
gradFuncFunction · 0.80
segOpComputeFunction · 0.80

Calls 1

runWebGLProgramMethod · 0.95

Tested by

no test coverage detected