Method profileKernel

tfjs-core/src/profiler.ts:40–79

(kernelName: string, inputs: NamedTensorMap, f: () => Tensor[])

  }

  profileKernel(kernelName: string, inputs: NamedTensorMap, f: () => Tensor[]):
      KernelProfile {
    let outputs: Tensor[];
    const holdResultWrapperFn = () => {
      outputs = f();
    };
    let timer: Promise<BackendTimingInfo>;
    const start = util.now();
    if (this.backendTimer.timerAvailable()) {
      timer = this.backendTimer.time(holdResultWrapperFn);
    } else {
      holdResultWrapperFn();
      for (const output of outputs) {
        output.dataSync();
      }
      timer = Promise.resolve({kernelMs: util.now() - start});
    }
    if (env().getBool('CHECK_COMPUTATION_FOR_ERRORS')) {
      for (let i = 0; i < outputs.length; i++) {
        const output = outputs[i];
        // Dangling promise here because we don't want to propagate up
        // asynchronicity.
        output.data().then(tensorVals => {
          checkComputationForErrors(tensorVals, output.dtype, kernelName);
        });
      }
    }

    const kernelProfile = {
      kernelName,
      outputs,
      inputs,
      timeMs: timer.then(timing => timing.kernelMs),
      extraInfo: timer.then(
          timing => timing.getExtraProfileInfo != null ?
              timing.getExtraProfileInfo() :
              '')
    };
    return kernelProfile;
  }

  logKernelProfile(kernelProfile: KernelProfile): void {
    const {kernelName, outputs, timeMs, inputs, extraInfo} = kernelProfile;