Searched refs:kReplicaId (Results 1 – 17 of 17) sorted by relevance
31 int id = kind() == Kind::kReplicaId ? logical_ids.first : logical_ids.second; in ExecuteOnStream()
42 : ReplicaOrPartitionIdThunk(Kind::kReplicaId, thunk_info, dest) {} in ReplicaIdThunk()
73 case Thunk::kReplicaId: in ThunkKindToString()
66 kReplicaId, enumerator
124 V(kReplicaId, "replica-id", 0) \
187 instruction->opcode() != HloOpcode::kReplicaId) { in Run()
111 if (hlo->opcode() == HloOpcode::kReplicaId) { in DetermineHloInstructionIsReplicated()
104 case HloOpcode::kReplicaId: in IsExpensive()
501 case HloOpcode::kReplicaId: { in CreateFromProto()1097 return absl::WrapUnique(new HloInstruction(HloOpcode::kReplicaId, shape)); in CreateReplicaId()1814 case HloOpcode::kReplicaId: in CloneWithNewOperands()2159 case HloOpcode::kReplicaId: in IdenticalSlowPath()3142 case HloOpcode::kReplicaId: in Visit()
1077 case HloOpcode::kReplicaId: in GetInstructionColor()
265 case HloOpcode::kReplicaId: in PickRepresentativeOperand()
136 case HloOpcode::kReplicaId: in CanInferShape()1333 case HloOpcode::kReplicaId: { in ParseInstructionRhs()
2305 case HloOpcode::kReplicaId: in InstructionCanChangeLayout()
330 return xla::HloOpcode::kReplicaId; in MhloToHloOpcode()
775 NoAttributeCase(kReplicaId, ReplicaIdOp); in ImportInstructionImpl()
354 case HloOpcode::kReplicaId: in EmitOp()
3022 return AddInstruction(std::move(instr), HloOpcode::kReplicaId, {}); in ReplicaId()