rafzi commented on issue #11881: URL: https://github.com/apache/tvm/issues/11881#issuecomment-1166344020
The following transformation seems to unroll all BatchMatMul ops to MatMul: https://github.com/tensorflow/tensorflow/blob/f72dafde88e0c32bc64144cdacc45b7b46d3c914/tensorflow/lite/toco/graph_transformations/unroll_batch_matmul.cc#L136 Using the MLIR converter switches from Slice to Split, but still does the unrolling:  So as far as I can tell it does not seem possible to generate the BatchMatMul op using the current converter. Maybe the model I referred to was created with an earlier version that did not employ this transformation? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
