Home
last modified time | relevance | path

Searched refs:infeed_data (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_domain_test.cc386 HloInstruction* infeed_data = FindInstruction(module.get(), "infeed.data"); in TEST_F() local
387 ASSERT_NE(infeed_data, nullptr); in TEST_F()
389 auto infeed_data_users = infeed_data->users(); in TEST_F()
390 HloInstruction* new_gte0 = infeed_data->parent()->AddInstruction( in TEST_F()
392 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 0), infeed_data, in TEST_F()
395 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
397 HloInstruction* new_gte1 = infeed_data->parent()->AddInstruction( in TEST_F()
399 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 1), infeed_data, in TEST_F()
402 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary( in TEST_F()
404 HloInstruction* new_tuple = infeed_data->parent()->AddInstruction( in TEST_F()
[all …]
Dhlo_dce_test.cc278 auto infeed_data = body_builder.AddInstruction( in TEST_F() local
281 shape, HloOpcode::kAdd, param, infeed_data)); in TEST_F()
Dlayout_assignment_test.cc791 auto infeed_data = false_builder.AddInstruction( in TEST_F() local
793 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
Dbuffer_assignment_test.cc2312 auto infeed_data = builder.AddInstruction( in TEST_F() local
2317 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2355 {token, infeed, infeed_data, while0, while1, zero, add, while2, tuple}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc267 auto infeed_data = std::make_shared<Literal>( in ReplayComputation() local
270 ->RegisterBeforeGetNextDestinationCallback([infeed_data, client] { in ReplayComputation()
271 TF_CHECK_OK(client->TransferToInfeed(*infeed_data)); in ReplayComputation()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1728 HloInstructionProto infeed_data; in Infeed() local
1729 *infeed_data.mutable_shape() = shape.ToProto(); in Infeed()
1730 infeed_data.set_tuple_index(0); in Infeed()
1731 return AddInstruction(std::move(infeed_data), HloOpcode::kGetTupleElement, in Infeed()