Home
last modified time | relevance | path

Searched refs:kOutfeed (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.cc51 case Thunk::kOutfeed: in ThunkKindToString()
Dthunk.h59 kOutfeed, enumerator
Doutfeed_thunk.cc28 : Thunk(Kind::kOutfeed, hlo_instruction), in OutfeedThunk()
Dir_emitter_unnested.cc1744 CHECK_EQ(HloOpcode::kOutfeed, inst->opcode()); in BuildOutfeedThunk()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h108 V(kOutfeed, "outfeed", 2) \
Dhlo_element_type_converter.cc132 opcode == HloOpcode::kInfeed || opcode == HloOpcode::kOutfeed) { in Run()
Dinstruction_fusion.cc89 case HloOpcode::kOutfeed: in IsExpensive()
Dhlo_instruction.cc367 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()
Dhlo_verifier.cc771 case HloOpcode::kOutfeed: in CheckMixedPrecisionOperands()
872 case HloOpcode::kOutfeed: in CheckShape()
Dhlo.proto108 // Outfeed configuration information, only present for kOutfeed.
Dlayout_assignment.cc450 } else if (instruction->opcode() == HloOpcode::kOutfeed) { in AddMandatoryConstraints()
2062 case HloOpcode::kOutfeed: in InstructionCanChangeLayout()
Dhlo_graph_dumper.cc1044 case HloOpcode::kOutfeed: in GetInstructionColor()
Dhlo_instructions.cc1816 : HloInstruction(HloOpcode::kOutfeed, ShapeUtil::MakeTokenShape()), in HloOutfeedInstruction()
Dhlo_parser.cc1408 case HloOpcode::kOutfeed: { in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc144 opcode == HloOpcode::kOutfeed || opcode == HloOpcode::kRng || in GetTargetParallelTaskCount()
Dir_emitter.h549 kOutfeed, enumerator
Dir_emitter.cc122 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/
Dreplay_computation.cc118 : HloOpcode::kOutfeed)) { in GetXfeedShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc230 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()