Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range_test.cc285 HloInstruction* body_iter = body_builder.AddInstruction( in TEST_F() local
293 scalar_shape, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_F()
327 {body_param, body_iter, body_data, body_iter_increment, in TEST_F()
Dmemory_space_assignment_test.cc763 HloInstruction* body_iter = body_builder.AddInstruction( in TEST_P() local
771 scalar_shape, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_P()
805 {body_param, body_iter, body_data, body_iter_increment, in TEST_P()
823 EXPECT_THAT(body_iter, op::ShapeWithLayout(scalar_shape)); in TEST_P()
2646 HloInstruction* body_iter = body_builder.AddInstruction( in TEST_P() local
2654 scalar_shape, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_P()
2694 {body_param, body_iter, body_data, body_iter_increment, in TEST_P()
2967 HloInstruction* body_iter = body_builder.AddInstruction( in TEST_P() local
2975 scalar_shape, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_P()
3023 {body_param, body_iter, body_data, body_iter_increment, in TEST_P()
[all …]
Dbuffer_assignment_test.cc798 HloInstruction* body_iter = body_builder.AddInstruction( in TEST_F() local
812 s32_, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_F()