Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier_test.cc555 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local
559 EXPECT_TRUE(ShapeUtil::Equal(new_while->shape(), flat_tuple)); in TEST_F()
561 new_while->while_body()->root_instruction()->shape(), flat_tuple)); in TEST_F()
563 new_while->while_body()->parameter_instruction(0)->shape(), flat_tuple)); in TEST_F()
565 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
641 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local
643 EXPECT_TRUE(ShapeUtil::Equal(new_while->shape(), new_while_shape)); in TEST_F()
645 new_while->while_body()->root_instruction()->shape(), new_while_shape)); in TEST_F()
647 new_while->while_body()->parameter_instruction(0)->shape(), in TEST_F()
650 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
[all …]
Dwhile_util.cc121 HloInstruction* new_while = containing_computation->AddInstruction( in MakeInstructionsLiveIn() local
129 new_while, while_instr->shape().tuple_shapes_size()))); in MakeInstructionsLiveIn()
144 result.new_while_instr = new_while; in MakeInstructionsLiveIn()
Dwhile_loop_simplifier.cc965 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables() local
971 while_op, convert_to_old_form(new_while))); in TryMergeInductionVariables()
975 return new_while; in TryMergeInductionVariables()