Home
last modified time | relevance | path

Searched refs:gte1 (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_simplifier_test.cc178 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
183 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
203 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
208 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
236 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
241 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
251 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
256 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
270 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
273 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
Dcopy_insertion_test.cc614 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildNestedBodyComputation() local
618 HloInstruction::CreateGetTupleElement(data_shape_, gte1, 0)); in BuildNestedBodyComputation()
627 HloInstruction::CreateGetTupleElement(data_shape_, gte1, 1)); in BuildNestedBodyComputation()
1111 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local
1116 HloInstruction::CreateBinary(gte1->shape(), HloOpcode::kAdd, gte1, gte2)); in TEST_F()
1178 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local
1183 HloInstruction::CreateBinary(gte1->shape(), HloOpcode::kAdd, gte1, gte2)); in TEST_F()
1478 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local
1483 HloInstruction::CreateBinary(gte1->shape(), HloOpcode::kAdd, gte1, gte2)); in TEST_F()
1584 auto gte1 = builder.AddInstruction( in TEST_F() local
[all …]
Dhlo_domain_test.cc647 HloInstruction* gte1 = in TEST_F() local
651 HloInstruction::CreateUnary(gte1->shape(), HloOpcode::kCopy, gte1)); in TEST_F()
683 EXPECT_TRUE(gte1->has_sharding()); in TEST_F()
684 EXPECT_EQ(HloSharding::Tuple(gte1->shape(), {HloSharding::AssignDevice(1), in TEST_F()
686 gte1->sharding()); in TEST_F()
Dtuple_points_to_analysis_test.cc660 auto gte1 = builder.AddInstruction( in TEST_F() local
667 {{constant, {}}, {gte1, {}}, {ccall, {1}}}); in TEST_F()
909 auto gte1 = builder.AddInstruction( in TEST_F() local
912 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
919 EXPECT_TRUE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {1}, gte1)); in TEST_F()
921 EXPECT_FALSE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte1)); in TEST_F()
932 auto gte1 = builder.AddInstruction( in TEST_F() local
942 data_shape, gte1, update, {starts})); in TEST_F()
948 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
Dbfloat16_propagation_test.cc218 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
220 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
522 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
524 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
532 EXPECT_TRUE(OutputsBF16(gte1)); in TEST_F()
568 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
574 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, gte1)); in TEST_F()
584 EXPECT_FALSE(OutputsBF16(gte1)); in TEST_F()
944 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
949 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte1)); in TEST_F()
[all …]
Dhlo_alias_analysis_test.cc157 auto gte1 = builder.AddInstruction( in TEST_F() local
160 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
230 auto gte1 = builder.AddInstruction( in TEST_F() local
236 HloInstruction::CreateUnary(scalar_shape_, HloOpcode::kNegate, gte1)); in TEST_F()
257 EXPECT_EQ(analysis.GetUniqueBufferAt(gte1), in TEST_F()
279 auto gte1 = builder.AddInstruction( in TEST_F() local
282 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
303 EXPECT_EQ(analysis.GetUniqueBufferAt(gte1), in TEST_F()
305 EXPECT_EQ(analysis.GetUniqueBufferAt(gte1), in TEST_F()
Dhlo_dataflow_analysis_test.cc158 auto gte1 = builder.AddInstruction( in TEST_P() local
161 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_P()
177 EXPECT_FALSE(analysis.ValueIsDefinedAt(gte1)); in TEST_P()
188 HloPosition{gte1, {}})); in TEST_P()
199 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
2100 auto gte1 = builder.AddInstruction( in TEST_F() local
2103 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
2110 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {1}, gte1)); in TEST_F()
2112 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte1)); in TEST_F()
2123 auto gte1 = builder.AddInstruction( in TEST_F() local
[all …]
Dsharding_propagation_test.cc761 auto* gte1 = FindInstruction(module.get(), "gte.1"); in TEST_P() local
762 ASSERT_NE(gte1, nullptr); in TEST_P()
763 EXPECT_THAT(gte1, op::Sharding("{{replicated}, {devices=[1,2,2,1]0,1,2,3}}")); in TEST_P()
769 EXPECT_THAT(gte1->sharding().tuple_elements()[0], in TEST_P()
771 EXPECT_THAT(gte1->sharding().tuple_elements()[1], in TEST_P()
776 {gte->sharding(), gte1->sharding().tuple_elements()[0], in TEST_P()
777 gte1->sharding().tuple_elements()[1], gte2->sharding()}) { in TEST_P()
Dlayout_assignment_test.cc773 auto gte1 = true_builder.AddInstruction( in TEST_F() local
776 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
Dbuffer_assignment_test.cc2596 auto gte1 = 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.cc3750 auto gte1 = 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.cc101 auto gte1 = computation->AddInstruction( in RunTest2D() local
104 TF_CHECK_OK(dot->ReplaceOperandWith(1, gte1)); in RunTest2D()
165 auto gte1 = computation->AddInstruction( in RunTest1D() local
168 TF_CHECK_OK(reshape->ReplaceOperandWith(0, gte1)); in RunTest1D()
Dbuffer_donation_test.cc237 HloInstruction* gte1 = builder.AddInstruction( in CreateTestModule() local
239 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in CreateTestModule()
Ddynamism_inference_test.cc145 auto gte1 = GetTupleElement(tuple, 1); in TEST_F() local
146 auto tuple_2 = Tuple(&b, {gte0, gte1}); in TEST_F()
Dtuple_test.cc440 auto gte1 = GetTupleElement(gte0, 1); in XLA_TEST_F() local
441 Add(gte1, ConstantR1<float>(&builder, {10.0, 11.0, 12.0})); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc636 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
637 Add(gte0, gte1); in TEST_F()
661 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
662 Add(gte0, gte1, {0}); in TEST_F()
704 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
705 Add(gte0, gte1, /*broadcast_dimensions=*/{0}); // f32[<=10, 15] in TEST_F()
725 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
727 Select(gte0, gte1, gte1); in TEST_F()
912 auto gte1 = GetTupleElement(p1, 0); in TEST_F() local
913 std::vector<XlaOp> input_operands = {gte0, gte1}; in TEST_F()
[all …]