चेतावनी: यह एपीआई हटा दिया गया है और प्रतिस्थापन के स्थिर होने के बाद TensorFlow के भविष्य के संस्करण में हटा दिया जाएगा।

NcclAllReduce

संग्रह की मदद से व्यवस्थित रहें अपनी प्राथमिकताओं के आधार पर, कॉन्टेंट को सेव करें और कैटगरी में बांटें.
सार्वजनिक अंतिम वर्ग NcclAllReduce

सभी इनपुट टेंसरों में कमी वाले एक टेंसर को आउटपुट करता है।

एक ही `shared_name के भीतर ऑप्स को पास किए गए सभी इनपुट टेंसरों में कमी वाले एक टेंसर को आउटपुट करता है।

ग्राफ़ का निर्माण इस प्रकार किया जाना चाहिए कि यदि एक ऑप शेयर_नाम मान `c` के साथ चलता है, तो `num_devices` ऑप्स साझा_नाम मान `c` के साथ चलेगा। ऐसा करने में विफलता के कारण ग्राफ़ निष्पादन पूरा नहीं हो पाएगा।

इनपुट: कमी डेटा के लिए इनपुट: सभी `num_devices` उपकरणों में कमी का मूल्य। कमी: प्रदर्शन करने के लिए कमी ऑपरेशन। num_devices: इस कमी में भाग लेने वाले उपकरणों की संख्या। साझा_नाम: पहचानकर्ता जो समान कमी के ऑप्स के बीच साझा करता है।

सार्वजनिक तरीके

आउटपुट <टी>
asOutput ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
स्थिर <टी संख्या फैली> NcclAllReduce <टी>
बनाने ( स्कोप गुंजाइश, ओपेरैंड <टी> इनपुट, स्ट्रिंग में कमी, लांग numDevices, स्ट्रिंग sharedName)
एक नया NcclAllReduce ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
आउटपुट <टी>

विरासत में मिली विधियां

सार्वजनिक तरीके

सार्वजनिक आउटपुट <टी> asOutput ()

टेंसर का प्रतीकात्मक हैंडल लौटाता है।

TensorFlow संचालन के लिए इनपुट एक अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।

सार्वजनिक स्थिर NcclAllReduce <टी> बनाने ( स्कोप गुंजाइश, ओपेरैंड <टी> इनपुट, स्ट्रिंग में कमी, लांग numDevices, स्ट्रिंग sharedName)

एक नया NcclAllReduce ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।

मापदंडों
दायरा वर्तमान दायरा
रिटर्न
  • NcclAllReduce का एक नया उदाहरण

सार्वजनिक आउटपुट <टी> डेटा ()