Searched refs:add_reduce_in_f32 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1186 auto add_reduce_in_f32 = [&](HloInstruction* hlo, const int64 dim) { in HandleDotStrengthReduction() local
1211 dot, reshape_if_necessary(add_reduce_in_f32(multiply(lhs, rhs), 0)))); in HandleDotStrengthReduction()
1243 ReplaceInstruction(dot, reshape_if_necessary(add_reduce_in_f32( in HandleDotStrengthReduction()
1248 dot, reshape_if_necessary(add_reduce_in_f32( in HandleDotStrengthReduction()
1263 dot, reshape_if_necessary(add_reduce_in_f32( in HandleDotStrengthReduction()
1333 ReplaceInstruction(dot, reshape_if_necessary(add_reduce_in_f32( in HandleDotStrengthReduction()
1344 dot, reshape_if_necessary(add_reduce_in_f32(multiply(new_lhs, rhs), in HandleDotStrengthReduction()
1355 dot, reshape_if_necessary(add_reduce_in_f32(multiply(lhs, new_rhs), in HandleDotStrengthReduction()
|