@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
17 lines (16 loc) • 653 B
TypeScript
import { Mode } from '../../model/module';
import Tensor, { DType } from '../../types';
import { OnnxNode } from '../node';
import { Attributes, Constants } from '../types';
export declare class BatchNormalizationNode extends OnnxNode {
private epsilon;
private momentum;
epsTensor: Tensor<any>;
constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode);
forward<DTpe extends DType>(inputs: Tensor<DTpe>[]): Promise<Tensor<DTpe>[]>;
getType(): string;
toCPU(): Promise<void>;
toWASM(): Promise<void>;
toGPU(): Promise<void>;
delete(): void;
}