UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

14 lines (13 loc) 553 B
import { Mode } from '../../model/module'; import Tensor from '../../types'; import { OnnxNode } from '../node'; import { Attributes, Constants } from '../types'; export declare class UpsampleNode extends OnnxNode { private sampleMode; private scales?; constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode); getScales(scale: Tensor<any>): Promise<number[]>; forward(inputs: Tensor<any>[]): Promise<Tensor<any>[]>; getType(): string; delete(): void; }