Shape inference in case of ops with complex element types need to use the element type of complex as the result element type and not the full operand type. Before: "mhlo.abs"(%arg0) : (tensor<4xcomplex<f32>>) -> tensor<4xtensor<4xcomplex<f32>>> After: "mhlo.abs"(%arg0) : (tensor<4xcomplex<f32>>) -> tensor<4xf32> PiperOrigin-RevId: 348123967 |
||
---|---|---|
.. | ||
Dialect | ||
utils | ||
CMakeLists.txt |