Searched refs:CreateOutfeed (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner_test.cc | 154 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 | 284 HloInstruction::CreateOutfeed(shape, param, token, "")); in TEST_F()
|
| D | hlo_instruction_test.cc | 722 HloInstruction::CreateOutfeed(shape10, constant, token, "")); in TEST_F()
724 HloInstruction::CreateOutfeed(shape01, constant, token, "")); in TEST_F()
|
| D | hlo_instruction.h | 543 static std::unique_ptr<HloInstruction> CreateOutfeed(
|
| D | hlo_instruction.cc | 371 instruction = CreateOutfeed(outfeed_shape, operands(0), operands(1), in CreateFromProto()
869 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateOutfeed( in CreateOutfeed() function in xla::HloInstruction
|
| D | hlo_parser.cc | 1416 HloInstruction::CreateOutfeed(operands[0]->shape(), operands[0], in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 655 HloInstruction::CreateOutfeed(dot_shape, dot, token, "no_config")); in CreateComputationForDotAddOutputFusionTest()
|