// TODO handle properly kernel with layer multiplier
auto transposed_tensor = mir::transposeTensor<0, 1, 3, 2>(kernel_tensor);
kernel = createOp<mir::ops::ConstantOp>(graph, transposed_tensor)->getOutput(0);
- result =
- createOp<mir::ops::DepthwiseConv2DOp>(graph, input, kernel, mir::Shape(strides),
- padding_before, padding_after, mir::DataFormat::NCHW)
- ->getOutput(0);
+ result = createOp<mir::ops::DepthwiseConv2DOp>(graph, input, kernel, strides, padding_before,
+ padding_after, mir::DataFormat::NCHW)
+ ->getOutput(0);
}
else
{
kernel_tensor = fixGroupedKernel(group, kernel_tensor);
kernel_tensor = mir::transposeTensor<3, 0, 1, 2>(kernel_tensor);
kernel = createOp<mir::ops::ConstantOp>(graph, kernel_tensor)->getOutput(0);
- result = createOp<mir::ops::Conv2DOp>(graph, input, kernel, mir::Shape(strides), padding_before,
+ result = createOp<mir::ops::Conv2DOp>(graph, input, kernel, strides, padding_before,
padding_after, mir::DataFormat::NCHW)
->getOutput(0);
}