Searched refs:CreateOutfeed (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner_test.cc | 194 HloInstruction::CreateOutfeed(f32, value, token, /*outfeed_config=*/"")); in TEST_F()
|
| D | while_loop_invariant_code_motion_test.cc | 269 HloInstruction::CreateOutfeed(scalar_s32, gte_0, in_token, "")); in TEST_F()
321 builder.AddInstruction(HloInstruction::CreateOutfeed( in TEST_F()
|
| D | hlo_dce_test.cc | 324 HloInstruction::CreateOutfeed(shape, param, token, "")); in TEST_F()
|
| D | hlo_instruction_test.cc | 724 HloInstruction::CreateOutfeed(shape10, constant, token, "")); in TEST_F()
726 HloInstruction::CreateOutfeed(shape01, constant, token, "")); in TEST_F()
|
| D | hlo_instruction.h | 742 static std::unique_ptr<HloInstruction> CreateOutfeed(
|
| D | hlo_instruction.cc | 408 instruction = CreateOutfeed(outfeed_shape, operands(0), operands(1), in CreateFromProto()
1115 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateOutfeed( in CreateOutfeed() function in xla::HloInstruction
|
| D | hlo_parser.cc | 2119 instruction = builder->AddInstruction(HloInstruction::CreateOutfeed( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 664 HloInstruction::CreateOutfeed(dot_shape, dot, token, "no_config")); in CreateComputationForDotAddOutputFusionTest()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2916 return b_.AddInstruction(HloInstruction::CreateOutfeed( in HandleOutfeed()
3028 branch_b.AddInstruction(HloInstruction::CreateOutfeed( in HandleOutfeed()
|