e6a1f5f0f9
This op is useful for rank specialization of broadcasts. Kernel Generator needs to generate one kernel for each rank, so if we can minimize the rank of the broadcast shape, we can support more cases with the same number of special-cased kernels. PiperOrigin-RevId: 360137827 |
||
|---|---|---|
| .. | ||
| 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 | ||