Searched refs:collective_op_kind (Results 1 – 3 of 3) sorted by relevance
84 CollectiveOpKind collective_op_kind, int64 op_id) in RendezvousKey()88 collective_op_kind(collective_op_kind), in RendezvousKey()95 static_cast<int>(k.collective_op_kind), k.op_id); in AbslHashValue()100 a.collective_op_kind == b.collective_op_kind && //112 num_local_participants, static_cast<int>(collective_op_kind), op_id); in ToString()118 CollectiveOpKind collective_op_kind; member
72 RendezvousKey::CollectiveOpKind collective_op_kind; member92 config.collective_op_kind = RendezvousKey::kCrossModule; in GetNcclCollectiveConfigForMlir()95 config.collective_op_kind = RendezvousKey::kCrossReplica; in GetNcclCollectiveConfigForMlir()
91 config().collective_op_kind, config().op_id); in ExecuteOnStream()