Home
last modified time | relevance | path

Searched refs:tuple_param (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_simplifier_test.cc172 HloInstruction* tuple_param = builder.AddInstruction( in TEST_F() local
175 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F()
177 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F()
179 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 2)); in TEST_F()
190 EXPECT_THAT(computation->root_instruction(), tuple_param); in TEST_F()
197 HloInstruction* tuple_param = builder.AddInstruction( in TEST_F() local
200 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F()
202 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F()
260 HloInstruction* tuple_param = builder.AddInstruction( in TEST_F() local
263 HloInstruction::CreateCall(tuple_shape_, {tuple_param}, c0)); in TEST_F()
[all …]
H A Dbuffer_assignment_test.cc1497 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
1515 tuple_param->shape(), in TEST_F()
1516 [this, &assignment, tuple_param](const Shape& /*subshape*/, in TEST_F()
1518 auto allocation = GetAllocation(*assignment, tuple_param, index); in TEST_F()
1529 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
1538 ShapeUtil::GetSubshape(tuple_param->shape(), {1}), tuple_param, 1)); in TEST_F()
1546 GetAllocation(*assignment, tuple_param, /*index=*/{}).maybe_live_out()); in TEST_F()
1550 GetAllocation(*assignment, tuple_param, /*index=*/{1}).maybe_live_out()); in TEST_F()
1551 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 0}) in TEST_F()
1553 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 1}) in TEST_F()
[all …]
H A Dhlo_dataflow_analysis_test.cc2141 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
2144 HloInstruction::CreateGetTupleElement(data_shape, tuple_param, 0)); in TEST_F()
2146 HloInstruction::CreateGetTupleElement(data_shape, tuple_param, 1)); in TEST_F()
2180 dataflow_analysis->DoesNotUseOperandBuffer(tuple_param, {1}, fusion)); in TEST_F()
2182 dataflow_analysis->DoesNotUseOperandBuffer(tuple_param, {0}, fusion)); in TEST_F()