Searched refs:reshape_if_necessary (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1177 auto reshape_if_necessary = [&](HloInstruction* hlo) { in HandleDotStrengthReduction() local
1211 dot, reshape_if_necessary(add_reduce_in_f32(multiply(lhs, rhs), 0)))); in HandleDotStrengthReduction()
1218 dot, reshape_if_necessary(multiply(Flatten(lhs), Flatten(rhs))))); 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()
1314 dot, reshape_if_necessary( 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()
|