Home
last modified time | relevance | path

Searched refs:gte1 (Results 1 – 13 of 13) 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.cc608 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildNestedBodyComputation() local
612 HloInstruction::CreateGetTupleElement(data_shape_, gte1, 0)); in BuildNestedBodyComputation()
621 HloInstruction::CreateGetTupleElement(data_shape_, gte1, 1)); in BuildNestedBodyComputation()
906 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local
911 HloInstruction::CreateBinary(gte1->shape(), HloOpcode::kAdd, gte1, gte2)); in TEST_F()
968 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local
973 HloInstruction::CreateBinary(gte1->shape(), HloOpcode::kAdd, gte1, gte2)); in TEST_F()
1268 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local
1273 HloInstruction::CreateBinary(gte1->shape(), HloOpcode::kAdd, gte1, gte2)); in TEST_F()
1374 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()
Dbuffer_liveness_test.cc626 auto gte1 = builder.AddInstruction( in BuildModule() local
635 HloInstruction::CreateSlice(update_shape, gte1, {0}, {3}, {1})); in BuildModule()
644 data_shape, gte1, update, {starts})); in BuildModule()
655 {dynamic_update_slice, starts, update, CHECK_NOTNULL(slice), gte1}, in BuildModule()
659 {dynamic_update_slice, starts, update, gte1}, in BuildModule()
784 auto gte1 = builder.AddInstruction( in Run() local
793 data_shape, HloOpcode::kAdd, gte0, gte1)); in Run()
800 data_shape, gte1, update, {starts})); in Run()
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()
260 EXPECT_EQ(analysis.GetUniqueBufferAt(gte1), in TEST_F()
282 auto gte1 = builder.AddInstruction( in TEST_F() local
285 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
309 EXPECT_EQ(analysis.GetUniqueBufferAt(gte1), in TEST_F()
311 EXPECT_EQ(analysis.GetUniqueBufferAt(gte1), in TEST_F()
Dhlo_dataflow_analysis_test.cc155 auto gte1 = builder.AddInstruction( in TEST_P() local
158 HloInstruction::CreateBinary(scalar_shape_, HloOpcode::kAdd, gte0, gte1)); in TEST_P()
174 EXPECT_FALSE(analysis.ValueIsDefinedAt(gte1)); in TEST_P()
185 HloPosition{gte1, {}})); in TEST_P()
196 UnorderedElementsAre(HloUse{gte0, 0, {}}, HloUse{gte1, 0, {}})); in TEST_P()
1974 auto gte1 = builder.AddInstruction( in TEST_F() local
1977 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
1984 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {1}, gte1)); in TEST_F()
1986 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte1)); in TEST_F()
1997 auto gte1 = builder.AddInstruction( in TEST_F() local
[all …]
Dbfloat16_propagation_test.cc217 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
219 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
492 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
494 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
502 EXPECT_TRUE(OutputsBF16(gte1)); in TEST_F()
538 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
544 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, gte1)); in TEST_F()
554 EXPECT_FALSE(OutputsBF16(gte1)); in TEST_F()
914 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
919 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte1)); in TEST_F()
Dtuple_points_to_analysis_test.cc858 auto gte1 = builder.AddInstruction( in TEST_F() local
861 HloInstruction::CreateBinary(elem_shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
868 EXPECT_TRUE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {1}, gte1)); in TEST_F()
870 EXPECT_FALSE(points_to_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte1)); in TEST_F()
881 auto gte1 = builder.AddInstruction( in TEST_F() local
891 data_shape, gte1, update, {starts})); in TEST_F()
897 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
975 auto gte1 = builder.AddInstruction( in TEST_F() local
985 data_shape, gte1, update, {starts})); in TEST_F()
991 {dynamic_update_slice, starts, update, gte1}, in TEST_F()
Dlayout_assignment_test.cc776 auto gte1 = true_builder.AddInstruction( in TEST_F() local
779 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, gte0, gte1)); in TEST_F()
Dbuffer_assignment_test.cc2414 auto gte1 = 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.cc538 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
539 Add(gte0, gte1); in TEST_F()
563 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
564 Add(gte0, gte1, {0}); in TEST_F()
606 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
607 Add(gte0, gte1, /*broadcast_dimensions=*/{0}); // f32[<=10, 15] in TEST_F()
627 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
629 Select(gte0, gte1, gte1); in TEST_F()
886 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local
887 Select(pred, gte0, gte1); in TEST_F()
[all …]
/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()
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()