Searched refs:while_hlo (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | while_transformer_test.cc | 102 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/ |
D | copy_insertion_test.cc | 656 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 …]
|
D | bfloat16_propagation.cc | 166 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()
|
D | bfloat16_propagation.h | 111 void DetermineWhileComputationsPrecision(HloInstruction* while_hlo);
|
D | bfloat16_propagation_test.cc | 629 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()
|
D | dynamic_dimension_inference_test.cc | 786 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()
|
D | hlo_evaluator.h | 241 Status HandleWhile(HloInstruction* while_hlo) override;
|
D | hlo_evaluator.cc | 2031 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()
|