@hoff97/tensor-js/dist/lib/onnx/optimizations/convRelu6.js

PyTorch like deep learning inferrence library

import { ConvNode } from '../nodes/conv/conv';
import { SequenceOptimization } from './optimization';
export class ConvRelu6 extends SequenceOptimization {
    constructor() {
        super(['Conv', 'Clip']);
    }
    apply(nodes, resolveConstant, constants, onnxVersion) {
        const conv = nodes[0];
        const clip = nodes[1];
        return new ConvNode(Object.entries(conv.attributes).map(x => x[1]), conv.inputs, clip.outputs, constants, onnxVersion, conv.mode, conv.kernel, conv.bias, 'relu6');
    }
    canApply(nodes) {
        const clip = nodes[1];
        return clip.min === 0 && clip.max === 6;
    }
}
//# sourceMappingURL=convRelu6.js.map