@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
14 lines (13 loc) • 553 B
TypeScript
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;
}