search

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.56k stars 569 forks source link

MobilenetV3 should compile with dynamic batch size #6357

Open stellaraccident opened 3 years ago

stellaraccident commented 3 years ago

Missing lowerings:

<unknown>:0: error: The following Tensorflow operations still remain: 
        tf.DepthwiseConv2dNative (count: 15)
        tf.Squeeze (count: 1)

Example:

%199 = "tf.DepthwiseConv2dNative"(%198, %cst_21) {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1]} : (tensor<?x14x14x184xf32>, tensor<3x3x184x1xf32>) -> tensor<?x14x14x184xf32>
%443 = "tf.Squeeze"(%442) {device = "", squeeze_dims = [1, 2]} : (tensor<?x1x1x1001xf32>) -> tensor<?x1001xf32>
jpienaar commented 2 years ago

Along which flow is this? (via MHLO I'm assuming but checking)

aaron-schneider commented 1 year ago

Ho there! This bug hasn't been updated in a long time. Good intentions and all, but we're moving this to the backlog. Feel free to bring it back if you think there's a reasonable chance it'll get worked on in the next 6mo!