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.cc1269 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
1287 tuple_param->shape(), in TEST_F()
1288 [this, &assignment, tuple_param](const Shape& /*subshape*/, in TEST_F()
1290 auto allocation = GetAllocation(*assignment, tuple_param, index); in TEST_F()
1301 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
1310 ShapeUtil::GetSubshape(tuple_param->shape(), {1}), tuple_param, 1)); in TEST_F()
1318 GetAllocation(*assignment, tuple_param, /*index=*/{}).maybe_live_out()); in TEST_F()
1322 GetAllocation(*assignment, tuple_param, /*index=*/{1}).maybe_live_out()); in TEST_F()
1323 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 0}) in TEST_F()
1325 EXPECT_TRUE(GetAllocation(*assignment, tuple_param, /*index=*/{1, 1}) in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc2029 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
2032 HloInstruction::CreateGetTupleElement(data_shape, tuple_param, 0)); in TEST_F()
2034 HloInstruction::CreateGetTupleElement(data_shape, tuple_param, 1)); in TEST_F()
2067 dataflow_analysis_->DoesNotUseOperandBuffer(tuple_param, {1}, fusion)); in TEST_F()
2069 dataflow_analysis_->DoesNotUseOperandBuffer(tuple_param, {0}, fusion)); in TEST_F()