Home
last modified time | relevance | path

Searched refs:broadcast_one_side (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2796 auto broadcast_one_side = [&](Value side, RankedTensorType type, in BroadcastBatchMatMulV2Operands() local
2813 broadcast_one_side(lhs, lhs_type, lhs_splitted.tail(), out_lhs); in BroadcastBatchMatMulV2Operands()
2814 broadcast_one_side(rhs, rhs_type, rhs_splitted.tail(), out_rhs); in BroadcastBatchMatMulV2Operands()