Home
last modified time | relevance | path

Searched refs:gte0 (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_simplifier_test.cc176 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
183 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
201 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
208 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
234 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
241 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
249 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
256 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
268 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
273 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
Dhlo_alias_analysis_test.cc155 auto gte0 = builder.AddInstruction( in TEST_F() local
160 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
181 analysis.GetUniqueBufferAt(gte0)); in TEST_F()
187 HloPosition{gte0, {}})); in TEST_F()
228 auto gte0 = builder.AddInstruction( in TEST_F() local
234 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte0)); in TEST_F()
254 EXPECT_EQ(analysis.GetUniqueBufferAt(gte0), in TEST_F()
277 auto gte0 = builder.AddInstruction( in TEST_F() local
282 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
298 EXPECT_EQ(analysis.GetUniqueBufferAt(gte0), in TEST_F()
[all …]
Dcopy_insertion_test.cc606 auto gte0 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildNestedBodyComputation() local
611 gte0->shape(), HloOpcode::kAdd, gte0, inc)); in BuildNestedBodyComputation()
748 auto gte0 = xla_while->parent()->AddInstruction( in BuildWhileInstruction_InitPointsToInterfering() local
752 HloInstruction::CreateTuple({gte0, sub})); in BuildWhileInstruction_InitPointsToInterfering()
1582 auto gte0 = builder.AddInstruction( in TEST_F() local
1586 builder.AddInstruction(HloInstruction::CreateTuple({gte1, gte0})); in TEST_F()
1615 auto gte0 = builder.AddInstruction( in TEST_F() local
1619 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
1648 auto gte0 = builder.AddInstruction( in TEST_F() local
1652 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
[all …]
Dtuple_points_to_analysis_test.cc658 auto gte0 = builder.AddInstruction( in TEST_F() local
665 ExpectHasBufferAliases(ccall, /*index=*/{0}, {{gte0, {}}, {ccall, {0}}}); in TEST_F()
907 auto gte0 = builder.AddInstruction( in TEST_F() local
912 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
918 EXPECT_TRUE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {0}, gte0)); in TEST_F()
920 EXPECT_FALSE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte0)); in TEST_F()
930 auto gte0 = builder.AddInstruction( in TEST_F() local
944 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
Dhlo_domain_test.cc644 HloInstruction* gte0 = in TEST_F() local
653 computation->AddInstruction(HloInstruction::CreateTuple({gte0, copy2})); in TEST_F()
680 EXPECT_TRUE(gte0->has_sharding()); in TEST_F()
681 EXPECT_EQ(HloSharding::AssignDevice(1), gte0->sharding()); in TEST_F()
Dbfloat16_propagation_test.cc216 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
220 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
520 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
524 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
531 EXPECT_TRUE(OutputsBF16(gte0)); in TEST_F()
566 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
572 ShapeUtil::MakeShape(F32, {4, 2}), gte0, {1, 0})); in TEST_F()
583 EXPECT_FALSE(OutputsBF16(gte0)); in TEST_F()
942 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
947 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte0)); in TEST_F()
Dhlo_dataflow_analysis_test.cc156 auto gte0 = builder.AddInstruction( in TEST_P() local
161 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_P()
176 EXPECT_FALSE(analysis.ValueIsDefinedAt(gte0)); in TEST_P()
184 HloPosition{gte0, {}})); in TEST_P()
199 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
2098 auto gte0 = builder.AddInstruction( in TEST_F() local
2103 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
2109 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {0}, gte0)); in TEST_F()
2111 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte0)); in TEST_F()
2121 auto gte0 = builder.AddInstruction( in TEST_F() local
[all …]
Dlayout_assignment_test.cc771 auto gte0 = true_builder.AddInstruction( in TEST_F() local
776 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
Dbuffer_assignment_test.cc2594 auto gte0 = builder.AddInstruction( in TEST_F() local
2599 HloInstruction::CreateBinary(data_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
2621 gte0, gte1, root_add}); in TEST_F()
Dmemory_space_assignment_test.cc3748 auto gte0 = builder.AddInstruction( in TEST_P() local
3753 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_P()
3758 schedule.set_sequence(computation, {param, gte0, gte1, tuple, while0}); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultioutput_fusion_test.cc99 auto gte0 = computation->AddInstruction( in RunTest2D() local
103 TF_CHECK_OK(dot->ReplaceOperandWith(0, gte0)); in RunTest2D()
163 auto gte0 = computation->AddInstruction( in RunTest1D() local
167 TF_CHECK_OK(sub->ReplaceOperandWith(0, gte0)); in RunTest1D()
Dbuffer_donation_test.cc235 HloInstruction* gte0 = builder.AddInstruction( in CreateTestModule() local
239 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in CreateTestModule()
Ddynamism_inference_test.cc144 auto gte0 = GetTupleElement(tuple, 0); in TEST_F() local
146 auto tuple_2 = Tuple(&b, {gte0, gte1}); in TEST_F()
Dtuple_test.cc439 auto gte0 = GetTupleElement(input, 0); in XLA_TEST_F() local
440 auto gte1 = GetTupleElement(gte0, 1); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc635 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
637 Add(gte0, gte1); in TEST_F()
660 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
662 Add(gte0, gte1, {0}); in TEST_F()
703 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
705 Add(gte0, gte1, /*broadcast_dimensions=*/{0}); // f32[<=10, 15] in TEST_F()
724 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
727 Select(gte0, gte1, gte1); in TEST_F()
911 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
913 std::vector<XlaOp> input_operands = {gte0, gte1}; in TEST_F()
[all …]