Searched refs:while_hlo1 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 900 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
907 ShapeUtil::GetSubshape(while_hlo1->shape(), {0}), while_hlo1, 0)); in TEST_F()
929 EXPECT_EQ(while_hlo1->operand(0)->operand(1)->opcode(), HloOpcode::kCopy); in TEST_F()
933 EXPECT_NE(while_hlo1->operand(0)->operand(1), in TEST_F()
962 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
969 ShapeUtil::GetSubshape(while_hlo1->shape(), {0}), while_hlo1, 0)); in TEST_F()
984 EXPECT_THAT(while_hlo1->operand(0), in TEST_F()
1262 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
1269 ShapeUtil::GetSubshape(while_hlo1->shape(), {0}), while_hlo1, 0)); in TEST_F()
1271 ShapeUtil::GetSubshape(while_hlo1->shape(), {0}), while_hlo2, 0)); in TEST_F()
[all …]
|