Searched refs:infeed_data (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_domain_test.cc | 386 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 …]
|
D | layout_assignment_test.cc | 794 auto infeed_data = false_builder.AddInstruction( in TEST_F() local 796 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
|
D | buffer_assignment_test.cc | 2164 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/ |
D | replay_computation.cc | 235 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/ |
D | xla_builder.cc | 1348 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()
|