Searched refs:CollectiveUidGenerator (Results 1 – 1 of 1) sorted by relevance
1097 class CollectiveUidGenerator { class1099 CollectiveUidGenerator() : cnt_(0) {} in CollectiveUidGenerator() function in tensorflow::CollectiveUidGenerator1267 explicit CollectiveOpLowering(MLIRContext* ctx, CollectiveUidGenerator& uid) in CollectiveOpLowering()1392 CollectiveUidGenerator& uid_;1432 CollectiveUidGenerator& uid) in AllReduceDoneOpLowering()1481 CollectiveUidGenerator& uid_;1599 CollectiveUidGenerator collective_uid; in runOnOperation()