Searched refs:kOutfeed (Results 1 – 19 of 19) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | thunk.cc | 51 case Thunk::kOutfeed: in ThunkKindToString()
|
D | thunk.h | 59 kOutfeed, enumerator
|
D | outfeed_thunk.cc | 28 : Thunk(Kind::kOutfeed, hlo_instruction), in OutfeedThunk()
|
D | ir_emitter_unnested.cc | 1744 CHECK_EQ(HloOpcode::kOutfeed, inst->opcode()); in BuildOutfeedThunk()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_opcode.h | 108 V(kOutfeed, "outfeed", 2) \
|
D | hlo_element_type_converter.cc | 132 opcode == HloOpcode::kInfeed || opcode == HloOpcode::kOutfeed) { in Run()
|
D | instruction_fusion.cc | 89 case HloOpcode::kOutfeed: in IsExpensive()
|
D | hlo_instruction.cc | 367 case HloOpcode::kOutfeed: { in CreateFromProto() 1245 case HloOpcode::kOutfeed: in HasSideEffectNoRecurse() 1376 case HloOpcode::kOutfeed: in CloneWithNewOperands() 1825 case HloOpcode::kOutfeed: in IdenticalSlowPath() 2627 case HloOpcode::kOutfeed: in Visit()
|
D | hlo_verifier.cc | 771 case HloOpcode::kOutfeed: in CheckMixedPrecisionOperands() 872 case HloOpcode::kOutfeed: in CheckShape()
|
D | hlo.proto | 108 // Outfeed configuration information, only present for kOutfeed.
|
D | layout_assignment.cc | 450 } else if (instruction->opcode() == HloOpcode::kOutfeed) { in AddMandatoryConstraints() 2062 case HloOpcode::kOutfeed: in InstructionCanChangeLayout()
|
D | hlo_graph_dumper.cc | 1044 case HloOpcode::kOutfeed: in GetInstructionColor()
|
D | hlo_instructions.cc | 1816 : HloInstruction(HloOpcode::kOutfeed, ShapeUtil::MakeTokenShape()), in HloOutfeedInstruction()
|
D | hlo_parser.cc | 1408 case HloOpcode::kOutfeed: { in ParseInstructionRhs()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_task_assignment.cc | 144 opcode == HloOpcode::kOutfeed || opcode == HloOpcode::kRng || in GetTargetParallelTaskCount()
|
D | ir_emitter.h | 549 kOutfeed, enumerator
|
D | ir_emitter.cc | 122 if (computation->root_instruction()->opcode() != HloOpcode::kOutfeed) { in EmitComputation() 486 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value); in HandleOutfeed() 497 TF_RETURN_IF_ERROR(EmitXfeedTransfer(XfeedKind::kOutfeed, in HandleOutfeed() 2677 if (root->opcode() == HloOpcode::kOutfeed) { in FinishVisit() 3096 if (root_inst->opcode() == HloOpcode::kOutfeed) { in GetBufferForGlobalCallReturnValue()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | replay_computation.cc | 118 : HloOpcode::kOutfeed)) { in GetXfeedShape()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 230 case HloOpcode::kOutfeed: in IsConstantVisitor() 1415 AddInstruction(std::move(instr), HloOpcode::kOutfeed, {operand, token}) in Outfeed() 1460 return AddInstruction(std::move(instr), HloOpcode::kOutfeed, in OutfeedWithToken()
|