HomeSort by relevance Sort by last modified time
    Searched defs:infeed_data (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_domain_test.cc 386 HloInstruction* infeed_data = FindInstruction(module.get(), "infeed.data"); local
387 ASSERT_NE(infeed_data, nullptr);
389 auto infeed_data_users = infeed_data->users();
390 HloInstruction* new_gte0 = infeed_data->parent()->AddInstruction(
392 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 0), infeed_data,
395 infeed_data->parent()->AddInstruction(HloInstruction::CreateUnary(
397 HloInstruction* new_gte1 = infeed_data->parent()->AddInstruction(
399 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 1), infeed_data,
    [all...]
layout_assignment_test.cc 794 auto infeed_data = false_builder.AddInstruction( local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/
replay_computation.cc 235 auto infeed_data = std::make_shared<Literal>( local
238 ->RegisterBeforeGetNextDestinationCallback([infeed_data, client] {
239 TF_CHECK_OK(client->TransferToInfeed(*infeed_data));

Completed in 163 milliseconds