Searched refs:kInfeed (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 43 case Thunk::kInfeed: in ThunkKindToString()
|
| D | thunk.h | 55 kInfeed, enumerator
|
| D | infeed_thunk.cc | 28 : Thunk(Kind::kInfeed, hlo_instruction), infeed_slices_(infeed_slices) {} in InfeedThunk()
|
| D | ir_emitter_unnested.cc | 1730 CHECK_EQ(HloOpcode::kInfeed, inst->opcode()); in BuildInfeedThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 94 V(kInfeed, "infeed", 1) \
|
| D | hlo_element_type_converter.cc | 132 opcode == HloOpcode::kInfeed || opcode == HloOpcode::kOutfeed) { in Run()
|
| D | layout_assignment.cc | 443 if (instruction->opcode() == HloOpcode::kInfeed) { in AddMandatoryConstraints()
1569 } else if (fused_instruction->opcode() == HloOpcode::kInfeed) { in SetFusionLayouts()
1691 if (instruction->opcode() != HloOpcode::kInfeed && in ClearComputationLayouts()
2060 case HloOpcode::kInfeed: in InstructionCanChangeLayout()
|
| D | instruction_fusion.cc | 79 case HloOpcode::kInfeed: in IsExpensive()
|
| D | hlo_instruction.cc | 358 case HloOpcode::kInfeed: { in CreateFromProto()
1244 case HloOpcode::kInfeed: in HasSideEffectNoRecurse()
1375 case HloOpcode::kInfeed: in CloneWithNewOperands()
1824 case HloOpcode::kInfeed: in IdenticalSlowPath()
2625 case HloOpcode::kInfeed: in Visit()
|
| D | hlo_verifier.cc | 770 case HloOpcode::kInfeed: in CheckMixedPrecisionOperands()
871 case HloOpcode::kInfeed: in CheckShape()
|
| D | hlo_graph_dumper.cc | 1043 case HloOpcode::kInfeed: in GetInstructionColor()
|
| D | hlo_instructions.cc | 1775 : HloInstruction(HloOpcode::kInfeed, in HloInfeedInstruction()
|
| D | hlo_parser.cc | 1387 case HloOpcode::kInfeed: { in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 143 opcode == HloOpcode::kFft || opcode == HloOpcode::kInfeed || in GetTargetParallelTaskCount()
|
| D | ir_emitter.h | 548 kInfeed, enumerator
|
| D | ir_emitter.cc | 369 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()
378 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
409 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
433 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
458 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 117 ? HloOpcode::kInfeed in GetXfeedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 229 case HloOpcode::kInfeed: in IsConstantVisitor()
1338 HloOpcode::kInfeed, {token})); in Infeed()
1341 HloOpcode::kInfeed, {token})); in Infeed()
1381 return AddInstruction(std::move(instr), HloOpcode::kInfeed, {token}); in InfeedWithToken()
|