Searched refs:kInfeed (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 63 case Thunk::kInfeed: in ThunkKindToString()
|
| D | infeed_thunk.cc | 31 : Thunk(Kind::kInfeed, thunk_info), dest_slices_(std::move(dest_slices)) {} in InfeedThunk()
|
| D | thunk.h | 58 kInfeed, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 96 V(kInfeed, "infeed", 1) \
|
| D | hlo_element_type_converter.cc | 132 opcode == HloOpcode::kInfeed || opcode == HloOpcode::kOutfeed) { in Run()
|
| D | hlo_replication_analysis.cc | 297 } else if (inst->opcode() == HloOpcode::kInfeed && cross_partition_spmd_) { in ComputeHloReplicationOnComputation()
|
| D | layout_assignment.cc | 459 if (instruction->opcode() == HloOpcode::kInfeed) { in AddMandatoryConstraints()
1710 } else if (fused_instruction->opcode() == HloOpcode::kInfeed) { in SetFusionLayouts()
1832 if (instruction->opcode() != HloOpcode::kInfeed && in ClearComputationLayouts()
2297 case HloOpcode::kInfeed: in InstructionCanChangeLayout()
|
| D | instruction_fusion.cc | 88 case HloOpcode::kInfeed: in IsExpensive()
|
| D | hlo_memory_scheduler.cc | 223 if (opcode == HloOpcode::kInfeed && !instruction->infeed_config().empty()) { in BytesFreedIfScheduled()
|
| D | hlo_instruction.cc | 395 case HloOpcode::kInfeed: { in CreateFromProto()
1530 case HloOpcode::kInfeed: in HasSideEffectNoRecurse()
1693 case HloOpcode::kInfeed: in CloneWithNewOperands()
2226 case HloOpcode::kInfeed: in IdenticalSlowPath()
3220 case HloOpcode::kInfeed: in Visit()
|
| D | sharding_propagation.cc | 256 case HloOpcode::kInfeed: in PickRepresentativeOperand()
309 case HloOpcode::kInfeed: in SupportSpatialPartitioning()
|
| D | hlo_verifier.cc | 1090 case HloOpcode::kInfeed: in CheckMixedPrecisionOperands()
1203 case HloOpcode::kInfeed: in CheckShape()
|
| D | hlo_graph_dumper.cc | 1070 case HloOpcode::kInfeed: in GetInstructionColor()
|
| D | hlo_parser.cc | 186 case HloOpcode::kInfeed: in CanInferShape()
2084 case HloOpcode::kInfeed: { in ParseInstructionRhs()
|
| D | hlo_instructions.cc | 2050 : HloInstruction(HloOpcode::kInfeed, in HloInfeedInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 574 kInfeed, enumerator
|
| D | ir_emitter.cc | 433 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()
442 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
467 kind == XfeedKind::kInfeed in EmitXfeedTransfer()
480 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
504 kind == XfeedKind::kInfeed in EmitXfeedTransfer()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 142 ? HloOpcode::kInfeed in GetXfeedShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 322 return xla::HloOpcode::kInfeed; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 381 case HloOpcode::kInfeed: { in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 345 case HloOpcode::kInfeed: in IsConstantVisitor()
1718 HloOpcode::kInfeed, {token})); in Infeed()
1721 HloOpcode::kInfeed, {token})); in Infeed()
1765 return AddInstruction(std::move(instr), HloOpcode::kInfeed, {token}); in InfeedWithTokenInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 310 case HloOpcode::kInfeed: in EmitOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 179 if (hlo->opcode() == HloOpcode::kInfeed) { in ClearShardingAttributes()
3740 hlo->opcode() == HloOpcode::kInfeed || in PreprocessSharding()
|