Class TestBackendTimer

tfjs-core/src/profiler_test.ts:26–43

import {TypedArray} from './types';

class TestBackendTimer implements BackendTimer {
  private counter = 1;
  constructor(
      private delayMs: number, private queryTimeMs: number,
      private extraInfo: string) {}

  timerAvailable() {
    return true;
  }

  async time(query: () => void): Promise<BackendTimingInfo> {
    query();
    const kernelMs = await new Promise<number>(
        resolve => setTimeout(
            () => resolve(this.queryTimeMs * this.counter++), this.delayMs));
    return {kernelMs, getExtraProfileInfo: () => this.extraInfo};
  }
}

class TestLogger extends Logger {
  override logKernelProfile(