UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

15 lines (14 loc) 565 B
import { Mode } from '../../model/module'; import Tensor from '../../types'; import { OnnxNode } from '../node'; import { Attributes, Constants } from '../types'; export declare class ConstantNode extends OnnxNode { tensor?: Tensor<any>; constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode); forward(inputs: Tensor<any>[]): Promise<Tensor<any>[]>; toCPU(): Promise<void>; toWASM(): Promise<void>; toGPU(): Promise<void>; getType(): string; delete(): void; }