Home
last modified time | relevance | path

Searched refs:while_hlo (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc102 auto while_hlo = builder.AddInstruction( in BuildWhileInstruction() local
106 return while_hlo; in BuildWhileInstruction()
130 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); in TEST_F() local
131 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
141 auto while_hlo = BuildWhileInstruction(condition, body, 1, 0); in TEST_F() local
142 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
152 auto while_hlo = BuildWhileInstruction(condition, body, 0, 10); in TEST_F() local
153 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
163 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); in TEST_F() local
164 auto result = ComputeWhileLoopTripCount(while_hlo); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion_test.cc656 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
659 return while_hlo; in BuildWhileInstruction()
664 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
667 return while_hlo; in BuildWhileInstruction()
772 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit() local
775 return while_hlo; in BuildWhileInstructionWithCustomInit()
804 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
813 EXPECT_THAT(while_hlo->operand(0), in TEST_F()
857 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
862 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
[all …]
Dbfloat16_propagation.cc166 HloInstruction* while_hlo) { in DetermineWhileComputationsPrecision() argument
167 CHECK_EQ(while_hlo->opcode(), HloOpcode::kWhile); in DetermineWhileComputationsPrecision()
174 HloComputation* body = while_hlo->while_body(); in DetermineWhileComputationsPrecision()
176 HloComputation* condition = while_hlo->while_condition(); in DetermineWhileComputationsPrecision()
179 body_root->shape(), [this, while_hlo, body_root]( in DetermineWhileComputationsPrecision()
184 if (OutputTypeAfterChange(while_hlo, index) == BF16) { in DetermineWhileComputationsPrecision()
189 << while_hlo->ToString(); in DetermineWhileComputationsPrecision()
Dbfloat16_propagation.h111 void DetermineWhileComputationsPrecision(HloInstruction* while_hlo);
Dbfloat16_propagation_test.cc629 auto while_hlo = builder.AddInstruction( in TEST_F() local
632 auto dot = builder.AddInstruction(CreateDot(shape, while_hlo, while_hlo)); in TEST_F()
695 auto while_hlo = builder.AddInstruction( in TEST_F() local
698 auto dot = builder.AddInstruction(CreateDot(shape, while_hlo, while_hlo)); in TEST_F()
772 auto while_hlo = builder.AddInstruction( in TEST_F() local
776 HloInstruction::CreateGetTupleElement(shape, while_hlo, 0)); in TEST_F()
778 HloInstruction::CreateGetTupleElement(shape, while_hlo, 1)); in TEST_F()
Ddynamic_dimension_inference_test.cc786 HloInstruction* while_hlo = nullptr; in TEST_F() local
790 while_hlo = inst; in TEST_F()
793 ASSERT_NE(while_hlo, nullptr); in TEST_F()
796 EXPECT_EQ(while_hlo->shape().tuple_shapes_size(), 4); in TEST_F()
798 for (HloInstruction* inst : while_hlo->while_body()->instructions()) { in TEST_F()
Dhlo_evaluator.h241 Status HandleWhile(HloInstruction* while_hlo) override;
Dhlo_evaluator.cc2031 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile() argument
2032 HloComputation* cond_comp = while_hlo->while_condition(); in HandleWhile()
2033 HloComputation* body_comp = while_hlo->while_body(); in HandleWhile()
2035 auto lcv = GetEvaluatedLiteralFor(while_hlo->operand(0)).Clone(); in HandleWhile()
2046 while_hlo->name(), max_loop_iterations_); in HandleWhile()
2060 evaluated_[while_hlo] = std::move(lcv); in HandleWhile()