de0578b4f9
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/40745 Fold broadcast_in_dim op if the operand is the result of a tensor splat. Copybara import of the project: -- 26c9f631448b8d6ffd20ece39ea8d4132b5550c7 by Uday Bondhugula <uday@polymagelabs.com>: [MLIR] Add constant folder for xla_hlo.broadcast_in_dim op Fold broadcast_in_dim op if the operand is the result of a tensor splat. COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/40745 from polymage-labs:broadcast_in_dim_fold 26c9f631448b8d6ffd20ece39ea8d4132b5550c7 PiperOrigin-RevId: 320365164 |
||
---|---|---|
include/mlir-hlo | ||
lib | ||
tests |