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()
|