| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| while_transformer_test.cc | 102 auto while_hlo = builder.AddInstruction( local
106 return while_hlo;
130 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); local
131 auto result = ComputeWhileLoopTripCount(while_hlo);
141 auto while_hlo = BuildWhileInstruction(condition, body, 1, 0); local
142 auto result = ComputeWhileLoopTripCount(while_hlo);
152 auto while_hlo = BuildWhileInstruction(condition, body, 0, 10); local
153 auto result = ComputeWhileLoopTripCount(while_hlo);
163 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); local
164 auto result = ComputeWhileLoopTripCount(while_hlo);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| bfloat16_propagation.h | 111 void DetermineWhileComputationsPrecision(HloInstruction* while_hlo);
|
| copy_insertion_test.cc | 650 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( local
653 return while_hlo;
658 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( local
661 return while_hlo;
766 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile local
798 auto while_hlo = BuildWhileInstruction(condition, body); local
826 auto while_hlo = BuildWhileInstruction(condition, body); local
1057 auto while_hlo = BuildWhileInstruction_InitPointsToConstant(); local
1074 auto while_hlo = BuildWhileInstruction_InitPointsToParameter(); local
1109 auto while_hlo = BuildWhileInstruction_InitPointsToAmbiguous(); local
1164 auto while_hlo = BuildWhileInstruction_InitPointsToNonDistinct(); local
1214 auto while_hlo = BuildWhileInstruction_InitPointsToInterfering(); local
[all...] |
| bfloat16_propagation.cc | 164 HloInstruction* while_hlo) {
165 CHECK_EQ(while_hlo->opcode(), HloOpcode::kWhile);
172 HloComputation* body = while_hlo->while_body();
174 HloComputation* condition = while_hlo->while_condition();
177 body_root->shape(), [this, while_hlo, body_root](
182 if (OutputTypeAfterChange(while_hlo, index) == BF16) {
187 << while_hlo->ToString();
[all...] |
| dynamic_dimension_inference_test.cc | 521 HloInstruction* while_hlo = nullptr;
525 while_hlo = inst;
528 ASSERT_NE(while_hlo, nullptr);
531 EXPECT_EQ(while_hlo->shape().tuple_shapes_size(), 3);
533 for (HloInstruction* inst : while_hlo->while_body()->instructions()) {
540 EXPECT_EQ(inference_->GetDynamicSize(while_hlo, {0}, 0), size_param);
541 EXPECT_EQ(inference_->GetDynamicSize(while_hlo, {1}, 0), size_param);
|
| bfloat16_propagation_test.cc | 599 auto while_hlo = builder.AddInstruction( local
602 auto dot = builder.AddInstruction(CreateDot(shape, while_hlo, while_hlo));
665 auto while_hlo = builder.AddInstruction( local
668 auto dot = builder.AddInstruction(CreateDot(shape, while_hlo, while_hlo));
742 auto while_hlo = builder.AddInstruction( local
[all...] |
| buffer_assignment.cc | 1562 const HloInstruction* while_hlo = instruction; local
[all...] |
| hlo_evaluator.h | 219 Status HandleWhile(HloInstruction* while_hlo) override;
|
| hlo_evaluator.cc | [all...] |
