mlir-hlo/lib/Dialect/mhlo
Adrian Kuegel 824bc9c425 Improve broadcast transformation to treat dynamic shapes with 1 element as scalar.
A shape that contains exactly one element is effectively a scalar. This leads
to a speedup in cases where we have a binary op with one operand that is
effectively a scalar, because we can use the fast path.

PiperOrigin-RevId: 357515552
2021-02-14 23:25:41 -08:00
..
IR Implement lowering of chlo::zeta to mhlo dialect. 2021-02-03 07:50:05 -08:00
transforms Improve broadcast transformation to treat dynamic shapes with 1 element as scalar. 2021-02-14 23:25:41 -08:00
CMakeLists.txt Add CMake files and lit configurations, enough for `ninja check-mlir-hlo` to pass on all the tests 2020-08-07 22:14:34 -07:00