Function checkKernelProfile

tfjs-core/src/profiler_test.ts:56–74

(acturalVal: KernelProfile, truthVal: {
  kernelName: string,
  outputs: Tensor[],
  timeMs: number|{error: string},
  inputs: NamedTensorMap,
  extraInfo: string
})

}

function checkKernelProfile(acturalVal: KernelProfile, truthVal: {
  kernelName: string,
  outputs: Tensor[],
  timeMs: number|{error: string},
  inputs: NamedTensorMap,
  extraInfo: string
}) {
  expect(acturalVal.kernelName).toBe(truthVal.kernelName);
  expect(acturalVal.inputs).toBe(truthVal.inputs);
  acturalVal.outputs.forEach((output, index) => {
    expect(output).toBe(truthVal.outputs[index]);
  });

  const promiseContainer = [
    promiseCheckWrapper(acturalVal.timeMs, truthVal.timeMs),
    promiseCheckWrapper(acturalVal.extraInfo, truthVal.extraInfo),
  ];
  return Promise.all(promiseContainer);
}

describeWithFlags('profiler.Profiler', SYNC_BACKEND_ENVS, () => {
  it('profiles simple function', doneFn => {