Searched defs:SPMDCollectiveOpsCreator (Results 1 – 1 of 1) sorted by relevance
125 struct SPMDCollectiveOpsCreator { struct127 std::function<HloInstruction*(SpmdBuilder*)> create_partition_id;134 create_cross_partition_all_reduce;159 // Create a default SPMDCollectiveOpsCreator. argument