@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
9 lines (8 loc) • 397 B
TypeScript
import { GPGPUProgram } from './gpgpu_math';
export declare class BatchNormProgram implements GPGPUProgram {
variableNames: string[];
outputShape: number[];
userCode: string;
supportsBroadcasting: boolean;
constructor(xShape: number[], meanShape: number[], varianceShape: number[], offsetShape: number[] | null, scaleShape: number[] | null, varianceEpsilon: number);
}