Searched refs:CommonPrefixBroadcast (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 476 static Value CommonPrefixBroadcast(Location loc, Value higher_rank_value, in CommonPrefixBroadcast() function
2728 CommonPrefixBroadcast(loc, logits, max_logits, rewriter); in matchAndRewrite()
2743 auto log_broadcast = CommonPrefixBroadcast(loc, logits, log, rewriter); in matchAndRewrite()
2747 auto sum_broadcast = CommonPrefixBroadcast(loc, logits, sum, rewriter); in matchAndRewrite()
|