@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
10 lines (9 loc) • 419 B
TypeScript
import Tensor from '../../types';
import { OnnxNode } from '../node';
import { Constants } from '../types';
import { SequenceOptimization } from './optimization';
export declare class ConvRelu6 extends SequenceOptimization {
constructor();
apply(nodes: OnnxNode[], resolveConstant: (name: string) => Tensor<any>, constants: Constants, onnxVersion: number): OnnxNode;
canApply(nodes: OnnxNode[]): boolean;
}