/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | tuple_simplifier_test.cc | 176 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 201 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 234 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 249 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 268 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local
|
H A D | copy_insertion_test.cc | 543 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildNestedBodyComputation() local 1013 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local 1080 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local 1324 auto gte1 = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() local 1430 auto gte1 = builder.AddInstruction( in TEST_F() local 1463 auto gte1 = builder.AddInstruction( in TEST_F() local 1496 auto gte1 = builder.AddInstruction( in TEST_F() local 1529 auto gte1 = builder.AddInstruction( in TEST_F() local 1564 auto gte1 = builder.AddInstruction( in TEST_F() local 1602 auto gte1 = builder.AddInstruction( in TEST_F() local
|
H A D | bfloat16_propagation_test.cc | 219 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 523 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 901 HloInstruction* gte1 = builder.AddInstruction( in TEST_F() local 1218 auto gte1 = FindInstruction(module.get(), "gte1"); in TEST_F() local
|
H A D | hlo_alias_analysis_test.cc | 190 auto gte1 = builder.AddInstruction( in TEST_F() local 263 auto gte1 = builder.AddInstruction( in TEST_F() local 312 auto gte1 = builder.AddInstruction( in TEST_F() local
|
H A D | hlo_dataflow_analysis_test.cc | 165 auto gte1 = builder.AddInstruction( in TEST_P() local 2083 auto gte1 = builder.AddInstruction( in TEST_F() local 2108 auto gte1 = builder.AddInstruction( in TEST_F() local 2152 auto gte1 = builder.AddInstruction( in TEST_F() local 2401 auto gte1 = builder.AddInstruction( in TEST_F() local 2440 auto gte1 = builder.AddInstruction( in TEST_F() local
|
H A D | hlo_domain_test.cc | 740 HloInstruction* gte1 = in TEST_F() local
|
H A D | memory_space_assignment_test.cc | 3711 auto gte1 = builder.AddInstruction( in TEST_P() local 4238 const HloInstruction* gte1 = FindInstruction(module.get(), "gte1"); in TEST_P() local 4365 const HloInstruction* gte1 = FindInstruction(module.get(), "gte1"); in TEST_P() local 4434 const HloInstruction* gte1 = FindInstruction(module.get(), "gte1"); in TEST_P() local
|
H A D | layout_assignment_test.cc | 736 auto gte1 = true_builder.AddInstruction( in TEST_F() local
|
H A D | hlo_evaluator_test.cc | 4926 HloInstruction* gte1 = in TEST_F() local 4958 HloInstruction* gte1 = in TEST_F() local
|
H A D | sharding_propagation_test.cc | 872 auto* gte1 = FindInstruction(module.get(), "gte.1"); in TEST_P() local
|
H A D | buffer_assignment_test.cc | 2735 auto gte1 = builder.AddInstruction( in TEST_F() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | xla_builder_test.cc | 706 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local 731 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local 774 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local 795 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local 1010 auto gte1 = GetTupleElement(p1, 0); in TEST_F() local 1124 auto gte1 = GetTupleElement(p0, 1); in TEST_F() local 1155 auto gte1 = GetTupleElement(p0, 1); // f32[4,5,<=6] in TEST_F() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | multioutput_fusion_test.cc | 99 auto gte1 = computation->AddInstruction( in RunTest2D() local 163 auto gte1 = computation->AddInstruction( in RunTest1D() local
|
H A D | buffer_donation_test.cc | 240 HloInstruction* gte1 = builder.AddInstruction( in CreateTestModule() local
|
H A D | tuple_test.cc | 280 auto gte1 = GetTupleElement(gte0, 1); in XLA_TEST_F() local
|
H A D | value_inference_test.cc | 114 auto gte1 = GetTupleElement(tuple, 1); in TEST_F() local
|