@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
9 lines (8 loc) • 322 B
TypeScript
import { ReduceInfo } from '../../ops/reduce_util';
import { GPGPUProgram } from './gpgpu_math';
export declare class ReduceProgram implements GPGPUProgram {
variableNames: string[];
outputShape: number[];
userCode: string;
constructor(reduceInfo: ReduceInfo, reduceType: 'max' | 'min' | 'sum');
}