| /external/tensorflow/tensorflow/compiler/xla/service/ |
| tuple_simplifier_test.cc | 176 HloInstruction* gte0 = builder.AddInstruction( local
183 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2}));
201 HloInstruction* gte0 = builder.AddInstruction( local
208 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1}));
234 HloInstruction* gte0 = builder.AddInstruction( local
241 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2}));
249 HloInstruction* gte0 = builder.AddInstruction( local
256 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2}));
268 HloInstruction* gte0 = builder.AddInstruction( local
273 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1}))
[all...] |
| hlo_domain_test.cc | 352 gte0 = f32[4] get-tuple-element(infeed.data), index=0
354 copy0 = f32[4] copy(gte0)
369 EXPECT_FALSE(HasDomainEdge(module.get(), "copy0", "gte0"));
379 // GTE0 GTE1
644 HloInstruction* gte0 = local
653 computation->AddInstruction(HloInstruction::CreateTuple({gte0, copy2}));
680 EXPECT_TRUE(gte0->has_sharding());
681 EXPECT_EQ(HloSharding::AssignDevice(1), gte0->sharding());
|
| buffer_liveness_test.cc | 623 auto gte0 = builder.AddInstruction( local
647 HloInstruction::CreateTuple({gte0, dynamic_update_slice}));
664 computation->CreateFusionInstruction({gte0},
781 auto gte0 = builder.AddInstruction( local
[all...] |
| hlo_alias_analysis_test.cc | 155 auto gte0 = builder.AddInstruction( local
160 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1));
181 analysis.GetUniqueBufferAt(gte0));
187 HloPosition{gte0, {}}));
228 auto gte0 = builder.AddInstruction( local
234 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0));
257 EXPECT_EQ(analysis.GetUniqueBufferAt(gte0),
280 auto gte0 = builder.AddInstruction( local
285 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1}));
304 EXPECT_EQ(analysis.GetUniqueBufferAt(gte0),
[all...] |
| layout_assignment_test.cc | 704 gte0 = f32[2,2,2] get-tuple-element(fparam), index=0
709 ROOT fresult = f32[2,2,2] add(gte0, add)
774 auto gte0 = true_builder.AddInstruction( local
[all...] |
| bfloat16_propagation_test.cc | 215 HloInstruction* gte0 = builder.AddInstruction( local
219 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1));
490 HloInstruction* gte0 = builder.AddInstruction( local
494 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1));
501 EXPECT_TRUE(OutputsBF16(gte0));
536 HloInstruction* gte0 = builder.AddInstruction( local
542 ShapeUtil::MakeShape(F32, {4, 2}), gte0, {1, 0}));
553 EXPECT_FALSE(OutputsBF16(gte0));
912 HloInstruction* gte0 = builder.AddInstruction( local
[all...] |
| copy_insertion_test.cc | 600 auto gte0 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( local
605 gte0->shape(), HloOpcode::kAdd, gte0, inc));
742 auto gte0 = xla_while->parent()->AddInstruction( local
746 HloInstruction::CreateTuple({gte0, sub}));
1372 auto gte0 = builder.AddInstruction( local
1407 auto gte0 = builder.AddInstruction( local
1442 auto gte0 = builder.AddInstruction( local
1475 auto gte0 = builder.AddInstruction( local
1511 auto gte0 = builder.AddInstruction( local
1550 auto gte0 = builder.AddInstruction( local
[all...] |
| tuple_points_to_analysis_test.cc | 856 auto gte0 = builder.AddInstruction( local
879 auto gte0 = builder.AddInstruction( local
973 auto gte0 = builder.AddInstruction( local
[all...] |
| hlo_dataflow_analysis_test.cc | 153 auto gte0 = builder.AddInstruction( local
158 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1));
173 EXPECT_FALSE(analysis.ValueIsDefinedAt(gte0));
181 HloPosition{gte0, {}}));
196 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}}));
1972 auto gte0 = builder.AddInstruction( local
1995 auto gte0 = builder.AddInstruction( local
2038 auto gte0 = builder.AddInstruction( local
2306 auto gte0 = builder.AddInstruction( local
2344 auto gte0 = builder.AddInstruction( local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| multioutput_fusion_test.cc | 99 auto gte0 = computation->AddInstruction( local
103 TF_CHECK_OK(dot->ReplaceOperandWith(0, gte0));
163 auto gte0 = computation->AddInstruction( local
167 TF_CHECK_OK(sub->ReplaceOperandWith(0, gte0));
237 gte0 = pred[4] get-tuple-element(fusion), index=0
240 ROOT select = f32[4] select(gte0, gte1, const)
262 gte0 = pred[] get-tuple-element(fusion), index=0
265 ROOT select = f32[] select(gte0, gte1, const)
|
| tuple_test.cc | 439 auto gte0 = GetTupleElement(input, 0); local
440 auto gte1 = GetTupleElement(gte0, 1);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder_test.cc | 537 auto gte0 = GetTupleElement(p0, 0); local
539 Add(gte0, gte1);
562 auto gte0 = GetTupleElement(p0, 0); local
564 Add(gte0, gte1, {0});
605 auto gte0 = GetTupleElement(p0, 0); local
607 Add(gte0, gte1, /*broadcast_dimensions=*/{0}); // f32[<=10, 15]
626 auto gte0 = GetTupleElement(p0, 0); local
629 Select(gte0, gte1, gte1);
826 auto gte0 = GetTupleElement(p0, 0); local
885 auto gte0 = GetTupleElement(p0, 0); local
916 auto gte0 = GetTupleElement(p0, 0); \/\/ f32[4,<=5,6] local
[all...] |
