XlaSpmdShardToFullShape

सार्वजनिक अंतिम कक्षा XlaSpmdShardToFullShape

मैन्युअल विभाजन से स्विच करने के लिए XLA SPMD पार्टीशनर द्वारा उपयोग किया जाने वाला एक ऑप

स्वचालित विभाजन. यह शार्ड-आकार, मैन्युअल रूप से विभाजित इनपुट को मैन्युअल विभाजन द्वारा उपयोग किए जाने वाले समान शार्डिंग के साथ स्वचालित रूप से विभाजित होने के लिए पूर्ण-आकार वाले टेंसर में परिवर्तित करता है।

स्थिरांक

डोरी OP_NAME इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है

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

आउटपुट <T>
आउटपुट के रूप में ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
स्थिर <T टीटाइप का विस्तार करता है > XlaSpmdShardToFullShape <T>
बनाएं ( स्कोप स्कोप, ऑपरेंड <टी> इनपुट, स्ट्रिंग मैनुअलशेयरिंग, शेप फुलशेप)
एक नए XlaSpmdShardToFullShape ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
आउटपुट <T>

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

स्थिरांक

सार्वजनिक स्थैतिक अंतिम स्ट्रिंग OP_NAME

इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है

स्थिर मान: "XlaSpmdShardToFullShape"

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

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

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

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

सार्वजनिक स्थैतिक XlaSpmdShardToFullShape <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, स्ट्रिंग मैनुअल शेयरिंग, शेप फुलशेप)

एक नए XlaSpmdShardToFullShape ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।

पैरामीटर
दायरा वर्तमान दायरा
रिटर्न
  • XlaSpmdShardToFullShape का एक नया उदाहरण

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