| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| fusion_merger_test.cc | 53 auto gte0 = builder.AddInstruction( local
66 data_shape_, HloOpcode::kAdd, one_vec, gte0));
119 auto gte0 = builder.AddInstruction( local
133 data_shape_, HloOpcode::kMultiply, gte0, gte1));
136 data_shape_, HloOpcode::kMultiply, gte0, mul0));
190 auto gte0 = builder.AddInstruction( local
199 HloInstruction::CreateBinary(data_shape_, HloOpcode::kAdd, gte0, gte1));
|
| instruction_fusion_test.cc | 146 auto gte0 = builder.AddInstruction( local
151 HloInstruction::CreateBinary(data_shape, HloOpcode::kAdd, gte0, gte1));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| tuple_simplifier_test.cc | 170 HloInstruction* gte0 = builder.AddInstruction( local
177 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2}));
195 HloInstruction* gte0 = builder.AddInstruction( local
202 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1}));
|
| liveness_util_test.cc | 60 auto gte0 = builder.AddInstruction( local
65 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1));
71 EXPECT_TRUE(DoesNotUseOperandBuffer(tuple, {0}, gte0, *points_to_analysis_));
73 EXPECT_FALSE(DoesNotUseOperandBuffer(tuple, {}, gte0, *points_to_analysis_));
76 EXPECT_TRUE(DoesNotUseOperandBuffer(tuple, {0}, gte0, *dataflow_analysis_));
78 EXPECT_FALSE(DoesNotUseOperandBuffer(tuple, {}, gte0, *dataflow_analysis_));
88 auto gte0 = builder.AddInstruction( local
102 HloInstruction::CreateTuple({gte0, dynamic_update_slice}));
202 auto gte0 = builder.AddInstruction( local
216 HloInstruction::CreateTuple({gte0, dynamic_update_slice}))
[all...] |
| buffer_liveness_test.cc | 622 auto gte0 = builder.AddInstruction( local
646 HloInstruction::CreateTuple({gte0, dynamic_update_slice}));
663 computation->CreateFusionInstruction({gte0},
752 auto gte0 = builder.AddInstruction( local
[all...] |
| while_loop_simplifier_test.cc | 326 auto* gte0 = body_builder.AddInstruction( local
329 body_builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte0}));
|
| hlo_alias_analysis_test.cc | 152 auto gte0 = builder.AddInstruction( local
157 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1));
177 analysis.GetUniqueBufferAt(gte0));
183 HloPosition{gte0, {}}));
[all...] |
| layout_assignment_test.cc | 601 gte0 = f32[2,2,2] get-tuple-element(fparam), index=0
606 ROOT fresult = f32[2,2,2] add(gte0, add)
645 EXPECT_THAT(fused_instr("gte0")->shape().layout().minor_to_major(),
681 auto gte0 = true_builder.AddInstruction( local
686 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, gte0, gte1));
[all...] |
| copy_insertion_test.cc | 593 auto gte0 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( local
598 gte0->shape(), HloOpcode::kAdd, gte0, inc));
731 auto gte0 = xla_while->parent()->AddInstruction( local
735 HloInstruction::CreateTuple({gte0, sub}));
[all...] |
| hlo_dataflow_analysis_test.cc | 151 auto gte0 = builder.AddInstruction( local
156 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1));
170 EXPECT_FALSE(analysis.ValueIsDefinedAt(gte0));
178 HloPosition{gte0, {}}));
193 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}}));
[all...] |
| buffer_assignment_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| multioutput_fusion_test.cc | 91 auto gte0 = computation->AddInstruction( local
95 TF_CHECK_OK(dot->ReplaceOperandWith(0, gte0));
151 auto gte0 = computation->AddInstruction( local
155 TF_CHECK_OK(sub->ReplaceOperandWith(0, gte0));
|
| tuple_test.cc | 439 auto gte0 = builder.GetTupleElement(input, 0); local
440 auto gte1 = builder.GetTupleElement(gte0, 1);
|
