Searched defs:SPMDCollectiveOpsCreator (Results 1 – 1 of 1) sorted by relevance
113 struct SPMDCollectiveOpsCreator { struct115 std::function<HloInstruction*(SpmdBuilder*)> create_partition_id;122 create_cross_partition_all_reduce;147 // Create a default SPMDCollectiveOpsCreator. argument