Searched refs:body_data_mul (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_live_range_test.cc | 297 HloInstruction* body_data_mul = in TEST_F() local
305 shape, HloOpcode::kAdd, body_data_add, body_data_mul)); in TEST_F()
328 body_iter_next, body_data_increment, body_data_mul, in TEST_F()
|
| D | memory_space_assignment_test.cc | 775 HloInstruction* body_data_mul = in TEST_P() local
783 shape, HloOpcode::kAdd, body_data_add, body_data_mul)); in TEST_P()
806 body_iter_next, body_data_increment, body_data_mul, in TEST_P()
830 EXPECT_THAT(body_data_mul, op::ShapeWithLayout(shape_in_alternate_mem)); in TEST_P()
2658 HloInstruction* body_data_mul = in TEST_P() local
2666 shape, HloOpcode::kAdd, body_data_add, body_data_mul)); in TEST_P()
2695 body_iter_next, body_data_increment, body_data_mul, in TEST_P()
|