8d051723c0
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 |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| chlo_canonicalize.td | ||
| chlo_ops.cc | ||
| hlo_ops.cc | ||
| hlo_ops_base_enums.cc | ||
| hlo_ops_base_structs.cc | ||
| hlo_patterns.td | ||
| infer_fusibility_op_interface.cc | ||
| init.cc | ||
| lhlo_gpu_ops.cc | ||
| lhlo_gpu_ops_enums.cc | ||
| lhlo_gpu_ops_structs.cc | ||
| lhlo_ops.cc | ||
| lhlo_ops_structs.cc | ||
| mhlo_canonicalize.td | ||