Searched refs:StripDim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 266 HloInstruction* StripDim(HloInstruction* hlo, int64 dim) { in StripDim() function in xla::__anondb6e78140111::AlgebraicSimplifierVisitor
1315 multiply(broadcast(StripDim(lhs, lhs_collapsing_dim), in HandleDotStrengthReduction()
1317 broadcast(StripDim(rhs, rhs_collapsing_dim), in HandleDotStrengthReduction()
1327 auto new_lhs = StripDim(lhs, lhs_kept_dim); in HandleDotStrengthReduction()
1328 auto new_rhs = StripDim(rhs, rhs_kept_dim); in HandleDotStrengthReduction()
1341 auto new_lhs = broadcast(StripDim(lhs, lhs_kept_dim), rhs->shape(), in HandleDotStrengthReduction()
1352 auto new_rhs = broadcast(StripDim(rhs, rhs_kept_dim), lhs->shape(), in HandleDotStrengthReduction()
|