Searched refs:CreateInfeed (Results 1 – 10 of 10) 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 | 169 false_computation->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
|
| D | hlo_dce_test.cc | 275 HloInstruction::CreateInfeed(shape, token, "")); in TEST_F()
|
| D | layout_assignment_test.cc | 753 HloInstruction::CreateInfeed(xshape, token, "")); in TEST_F()
|
| D | hlo_instruction.h | 840 static std::unique_ptr<HloInstruction> CreateInfeed(
|
| D | hlo_instruction.cc | 459 CreateInfeed(data_shape, operands(0), proto.infeed_config()); in CreateFromProto()
1382 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateInfeed( in CreateInfeed() function in xla::HloInstruction
|
| D | buffer_assignment_test.cc | 2389 builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, token, "")); in TEST_F()
|
| D | hlo_evaluator_test.cc | 4877 HloInstruction* infeed = b.AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
|
| D | hlo_parser.cc | 2451 return builder->AddInstruction(HloInstruction::CreateInfeed( in CreateInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3257 HloInstruction::CreateInfeed(shape, token, hlo->infeed_config())); in HandleInfeed()
3265 return b_.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed()
3326 auto infeed = branch_b.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed()
|