Searched refs:CreateOutfeed (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner_test.cc | 193 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 | 322 HloInstruction::CreateOutfeed(shape, param, token, "")); in TEST_F()
|
| D | hlo_instruction_test.cc | 823 HloInstruction::CreateOutfeed(shape10, constant, token, "")); in TEST_F()
825 HloInstruction::CreateOutfeed(shape01, constant, token, "")); in TEST_F()
|
| D | hlo_instruction.h | 847 static std::unique_ptr<HloInstruction> CreateOutfeed(
|
| D | hlo_instruction.cc | 465 instruction = CreateOutfeed(outfeed_shape, operands(0), operands(1), in CreateFromProto()
1389 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateOutfeed( in CreateOutfeed() function in xla::HloInstruction
|
| D | hlo_parser.cc | 2470 return builder->AddInstruction(HloInstruction::CreateOutfeed( in CreateInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 646 HloInstruction::CreateOutfeed(dot_shape, dot, token, "no_config")); in CreateComputationForDotAddOutputFusionTest()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3697 return b_.AddInstruction(HloInstruction::CreateOutfeed( in HandleOutfeed()
3810 branch_b.AddInstruction(HloInstruction::CreateOutfeed( in HandleOutfeed()
|