स्वचालित विभाजन से स्विच करने के लिए XLA SPMD पार्टीशनर द्वारा उपयोग किया जाने वाला एक ऑप
मैन्युअल विभाजन. यह मैन्युअल विभाजन द्वारा उपयोग किए जाने वाले समान शार्डिंग के साथ इनपुट (पूर्ण आकार, स्वचालित रूप से विभाजित होने के लिए) को एनोटेट करता है, और बाद में मैन्युअल रूप से विभाजित ऑप्स द्वारा उपभोग किए जाने वाले शार्ड-आकार वाले टेंसर को आउटपुट करता है। यदि आकृति समान रूप से विभाजन योग्य नहीं है, तो पैडिंग क्षेत्र को 0s से छुपा दिया जाएगा।
स्थिरांक
डोरी | OP_NAME | इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है |
सार्वजनिक तरीके
आउटपुट <T> | आउटपुट के रूप में () टेंसर का प्रतीकात्मक हैंडल लौटाता है। |
स्थिर <T टीटाइप का विस्तार करता है > XlaSpmdFullToShardShape <T> | |
आउटपुट <T> | आउटपुट () |
विरासत में मिली विधियाँ
स्थिरांक
सार्वजनिक स्थैतिक अंतिम स्ट्रिंग OP_NAME
इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है
सार्वजनिक तरीके
सार्वजनिक आउटपुट <T> asOutput ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
TensorFlow संचालन के इनपुट किसी अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।
सार्वजनिक स्थैतिक XlaSpmdFullToShardShape <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, स्ट्रिंग मैनुअलशेयरिंग)
एक नए XlaSpmdFullToShardShape ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|
रिटर्न
- XlaSpmdFullToShardShape का एक नया उदाहरण