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