Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_simplifier_test.cc174 HloInstruction* tuple_param = builder.AddInstruction( in TEST_F() local
177 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F()
179 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F()
181 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 2)); in TEST_F()
192 EXPECT_THAT(computation->root_instruction(), tuple_param); in TEST_F()
199 HloInstruction* tuple_param = builder.AddInstruction( in TEST_F() local
202 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F()
204 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F()
262 HloInstruction* tuple_param = builder.AddInstruction( in TEST_F() local
265 HloInstruction::CreateCall(tuple_shape_, {tuple_param}, c0)); in TEST_F()
[all …]
Dbuffer_assignment_test.cc1441 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
1459 tuple_param->shape(), in TEST_F()
1460 [this, &assignment, tuple_param](const Shape& /*subshape*/, in TEST_F()
1462 auto allocation = GetAllocation(*assignment, tuple_param, index); in TEST_F()
1473 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
1482 ShapeUtil::GetSubshape(tuple_param->shape(), {1}), tuple_param, 1)); in TEST_F()
1490 GetAllocation(*assignment, tuple_param, /*index=*/{}).maybe_live_out()); in TEST_F()
1494 GetAllocation(*assignment, tuple_param, /*index=*/{1}).maybe_live_out()); in TEST_F()
1495 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 0}) in TEST_F()
1497 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 1}) in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc2155 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
2158 HloInstruction::CreateGetTupleElement(data_shape, tuple_param, 0)); in TEST_F()
2160 HloInstruction::CreateGetTupleElement(data_shape, tuple_param, 1)); in TEST_F()
2193 dataflow_analysis_->DoesNotUseOperandBuffer(tuple_param, {1}, fusion)); in TEST_F()
2195 dataflow_analysis_->DoesNotUseOperandBuffer(tuple_param, {0}, fusion)); in TEST_F()