UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

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