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

PyTorch like deep learning inferrence library

import { ConvNode } from '../nodes/conv/conv';
import { SequenceOptimization } from './optimization';
export class ConvRelu extends SequenceOptimization {
    constructor() {
        super(['Conv', 'Relu']);
    }
    apply(nodes, resolveConstant, constants, onnxVersion) {
        const conv = nodes[0];
        const relu = nodes[1];
        return new ConvNode(Object.entries(conv.attributes).map(x => x[1]), conv.inputs, relu.outputs, constants, onnxVersion, conv.mode, conv.kernel, conv.bias, 'relu');
    }
}
//# sourceMappingURL=convRelu.js.map