@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
13 lines (12 loc) • 638 B
TypeScript
import { Mode } from '../../../model/module';
import Tensor, { DType } from '../../../types';
import { OnnxNode } from '../../node';
import { Attributes, Constants } from '../../types';
export declare abstract class BinaryNode extends OnnxNode {
protected name: string;
constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, name: string, mode: Mode);
abstract compute<DTpe extends DType>(a: Tensor<DTpe>, b: Tensor<DTpe>): Tensor<DTpe>;
forward<DTpe extends DType>(inputs: Tensor<DTpe>[]): Promise<Tensor<DTpe>[]>;
getType(): string;
delete(): void;
}