Home
last modified time | relevance | path

Searched refs:cond_iter (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range_test.cc272 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F() local
277 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_F()
325 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_F()
Dheap_simulator_test.cc52 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F() local
58 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_F()
92 {cond_param, cond_iter, cond_data, cond_lt}); in TEST_F()
895 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F() local
900 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_F()
919 {param, while_op, body_param, cond_param, cond_iter, cond_data, cond_lt}); in TEST_F()
Dmemory_space_assignment_test.cc748 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_P() local
754 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_P()
803 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
824 EXPECT_THAT(cond_iter, op::ShapeWithLayout(scalar_shape)); in TEST_P()
2631 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_P() local
2637 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_P()
2692 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
2954 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_P() local
2959 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_P()
3021 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
[all …]
Dbuffer_assignment_test.cc785 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F() local
790 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_F()