Home
last modified time | relevance | path

Searched refs:gte0 (Results 1 – 13 of 13) 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()
257 EXPECT_EQ(analysis.GetUniqueBufferAt(gte0), in TEST_F()
280 auto gte0 = builder.AddInstruction( in TEST_F() local
285 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
304 EXPECT_EQ(analysis.GetUniqueBufferAt(gte0), in TEST_F()
[all …]
Dcopy_insertion_test.cc600 auto gte0 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildNestedBodyComputation() local
605 gte0->shape(), HloOpcode::kAdd, gte0, inc)); in BuildNestedBodyComputation()
742 auto gte0 = xla_while->parent()->AddInstruction( in BuildWhileInstruction_InitPointsToInterfering() local
746 HloInstruction::CreateTuple({gte0, sub})); in BuildWhileInstruction_InitPointsToInterfering()
1372 auto gte0 = builder.AddInstruction( in TEST_F() local
1376 builder.AddInstruction(HloInstruction::CreateTuple({gte1, gte0})); in TEST_F()
1407 auto gte0 = builder.AddInstruction( in TEST_F() local
1411 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
1442 auto gte0 = builder.AddInstruction( in TEST_F() local
1446 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
[all …]
Dbuffer_liveness_test.cc623 auto gte0 = builder.AddInstruction( in BuildModule() local
647 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in BuildModule()
664 computation->CreateFusionInstruction({gte0}, in BuildModule()
781 auto gte0 = builder.AddInstruction( in Run() local
792 gte0 = builder.AddInstruction(HloInstruction::CreateBinary( in Run()
793 data_shape, HloOpcode::kAdd, gte0, gte1)); in Run()
803 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in Run()
Dbfloat16_propagation_test.cc215 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
219 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
490 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
494 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
501 EXPECT_TRUE(OutputsBF16(gte0)); in TEST_F()
536 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
542 ShapeUtil::MakeShape(F32, {4, 2}), gte0, {1, 0})); in TEST_F()
553 EXPECT_FALSE(OutputsBF16(gte0)); in TEST_F()
912 HloInstruction* gte0 = builder.AddInstruction( in TEST_F() local
917 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte0)); 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()
Dhlo_dataflow_analysis_test.cc153 auto gte0 = builder.AddInstruction( in TEST_P() local
158 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_P()
173 EXPECT_FALSE(analysis.ValueIsDefinedAt(gte0)); in TEST_P()
181 HloPosition{gte0, {}})); in TEST_P()
196 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
1972 auto gte0 = builder.AddInstruction( in TEST_F() local
1977 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
1983 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {0}, gte0)); in TEST_F()
1985 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte0)); in TEST_F()
1995 auto gte0 = builder.AddInstruction( in TEST_F() local
[all …]
Dtuple_points_to_analysis_test.cc856 auto gte0 = builder.AddInstruction( in TEST_F() local
861 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
867 EXPECT_TRUE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {0}, gte0)); in TEST_F()
869 EXPECT_FALSE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte0)); in TEST_F()
879 auto gte0 = builder.AddInstruction( in TEST_F() local
893 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
973 auto gte0 = builder.AddInstruction( in TEST_F() local
987 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in TEST_F()
Dlayout_assignment_test.cc774 auto gte0 = true_builder.AddInstruction( in TEST_F() local
779 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
Dbuffer_assignment_test.cc2412 auto gte0 = builder.AddInstruction( in TEST_F() local
2417 HloInstruction::CreateBinary(data_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
2439 gte0, gte1, root_add}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc537 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
539 Add(gte0, gte1); in TEST_F()
562 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
564 Add(gte0, gte1, {0}); in TEST_F()
605 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
607 Add(gte0, gte1, /*broadcast_dimensions=*/{0}); // f32[<=10, 15] in TEST_F()
626 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
629 Select(gte0, gte1, gte1); in TEST_F()
826 auto gte0 = GetTupleElement(p0, 0); in TEST_F() local
828 SelectAndScatter(gte0, ge, {1, 2, 4}, {1, 2, 4}, Padding::kValid, source, in TEST_F()
[all …]
/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()
Dtuple_test.cc439 auto gte0 = GetTupleElement(input, 0); in XLA_TEST_F() local
440 auto gte1 = GetTupleElement(gte0, 1); in XLA_TEST_F()