Home
last modified time | relevance | path

Searched refs:infeed_data (Results 1 – 5 of 5) 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 …]
Dlayout_assignment_test.cc794 auto infeed_data = false_builder.AddInstruction( in TEST_F() local
796 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
Dbuffer_assignment_test.cc2164 auto infeed_data = builder.AddInstruction( in TEST_F() local
2169 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2207 {token, infeed, infeed_data, while0, while1, zero, add, while2, tuple}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc235 auto infeed_data = std::make_shared<Literal>( in ReplayComputation() local
238 ->RegisterBeforeGetNextDestinationCallback([infeed_data, client] { in ReplayComputation()
239 TF_CHECK_OK(client->TransferToInfeed(*infeed_data)); in ReplayComputation()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1348 HloInstructionProto infeed_data; in Infeed() local
1349 *infeed_data.mutable_shape() = shape.ToProto(); in Infeed()
1350 infeed_data.set_tuple_index(0); in Infeed()
1351 return AddInstruction(std::move(infeed_data), HloOpcode::kGetTupleElement, in Infeed()