Searched refs:CreateInfeed (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_simplifier_test.cc | 221 while_body->AddInstruction(HloInstruction::CreateInfeed( in TEST_F() 236 while_cond->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
|
D | conditional_simplifier_test.cc | 170 false_computation->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
|
D | hlo_dce_test.cc | 277 HloInstruction::CreateInfeed(shape, token, "")); in TEST_F()
|
D | hlo_instruction.h | 735 static std::unique_ptr<HloInstruction> CreateInfeed(
|
D | layout_assignment_test.cc | 790 HloInstruction::CreateInfeed(xshape, token, "")); in TEST_F()
|
D | hlo_instruction.cc | 402 CreateInfeed(data_shape, operands(0), proto.infeed_config()); in CreateFromProto() 1108 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateInfeed( in CreateInfeed() function in xla::HloInstruction
|
D | buffer_assignment_test.cc | 2311 builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, token, "")); in TEST_F()
|
D | hlo_parser.cc | 2100 instruction = builder->AddInstruction(HloInstruction::CreateInfeed( in ParseInstructionRhs()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner.cc | 2502 HloInstruction::CreateInfeed(shape, token, hlo->infeed_config())); in HandleInfeed() 2510 return b_.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed() 2570 auto infeed = branch_b.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed()
|