Home
last modified time | relevance | path

Searched refs:kPartitionId (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreplica_id_thunk.h48 : ReplicaOrPartitionIdThunk(Kind::kPartitionId, thunk_info, dest) {} in PartitionIdThunk()
Dthunk.cc75 case Thunk::kPartitionId: in ThunkKindToString()
Dthunk.h67 kPartitionId, enumerator
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h114 V(kPartitionId, "partition-id", 0) \
Dhlo_cse.cc186 instruction->opcode() != HloOpcode::kPartitionId && in Run()
Dhlo_replication_analysis.cc115 if (hlo->opcode() == HloOpcode::kPartitionId) { in DetermineHloInstructionIsReplicated()
Dinstruction_fusion.cc100 case HloOpcode::kPartitionId: in IsExpensive()
Dhlo_instruction.cc505 case HloOpcode::kPartitionId: { in CreateFromProto()
1105 return absl::WrapUnique(new HloInstruction(HloOpcode::kPartitionId, shape)); in CreatePartitionId()
1818 case HloOpcode::kPartitionId: in CloneWithNewOperands()
2152 case HloOpcode::kPartitionId: in IdenticalSlowPath()
3144 case HloOpcode::kPartitionId: in Visit()
Dhlo_graph_dumper.cc1072 case HloOpcode::kPartitionId: in GetInstructionColor()
Dsharding_propagation.cc260 case HloOpcode::kPartitionId: in PickRepresentativeOperand()
Dhlo_parser.cc130 case HloOpcode::kPartitionId: in CanInferShape()
1341 case HloOpcode::kPartitionId: { in ParseInstructionRhs()
Dlayout_assignment.cc2301 case HloOpcode::kPartitionId: in InstructionCanChangeLayout()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc336 case HloOpcode::kPartitionId: in EmitOp()