You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Describe the bug
When translating the DepthwiseConv2d or DepthwiseConv2dNative operations 2 extra ops are added to the kernel to get into the correct form. These are the Reshape and Transpose operators. These are problematic for the following reasons.
The Transpose order (3, 2, 0, 1) is invalid under certain versions of tensorrt
The Reshape operator is invalid under certain versions of tensorrt (5.1 does not support the int64)
Additional complexity with 2 extra layers per DepthwiseConv2d operator
Urgency
This is urgent and blocker for anybody using the onnx system as a intermediate step towards other runtimes
System information
Linux Mint 18.04
Tensorflow Version: 1.15
Python version: 3.6.9
To Reproduce
Create any DepthwiseConv2dNative or DepthwiseConv2d operator in tensorflow. I use Opset 7 but this is true for any version
Expected behavior
I expected the kernel to be changed/reshaped internally and not use additional operators
Screenshots
Tensorflow
Onnx
Additional context
My target system is tensorrt 5.1 and tensorrt 7.0