सार्वजनिक अंतिम वर्ग रेप्लिकेटमेटाडेटा
मेटाडेटा यह दर्शाता है कि टीपीयू गणना को कैसे दोहराया जाना चाहिए।
यह ऑपरेशन `tpu.replication()` कंप्यूटेशन सबग्राफ के संचालन के लिए सामान्य मेटाडेटा रखता है।
नेस्टेड क्लासेस
कक्षा | प्रतिकृतिमेटाडेटा.विकल्प | ReplicateMetadata के लिए वैकल्पिक विशेषताएँ |
स्थिरांक
डोरी | OP_NAME | इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है |
सार्वजनिक तरीके
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | अनुमतिसॉफ्टप्लेसमेंट (बूलियन अनुमतिसॉफ्टप्लेसमेंट) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | संगणना आकार (सूची<लंबा> संगणना आकार) |
स्टेटिक रिप्लिकेटमेटाडेटा | |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | डिवाइसअसाइनमेंट (सूची<लंबी> डिवाइसअसाइनमेंट) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | होस्टकंप्यूटकोर (सूची<स्ट्रिंग>होस्टकंप्यूटकोर) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | numCoresPerReplika (लंबा numCoresPerReplusa) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | पैडिंगमैप (सूची<स्ट्रिंग> पैडिंगमैप) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | स्टेपमार्करलोकेशन (स्ट्रिंग स्टेपमार्करलोकेशन) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | टोपोलॉजी (स्ट्रिंग टोपोलॉजी) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | यूज़SpmdForXlaPartitioning (बूलियन यूज़SpmdForXlaPartitioning) |
स्थैतिक रेप्लिकेटमेटाडेटा.विकल्प | यूज़ टीपीयू (बूलियन यूज़ टीपीयू) |
विरासत में मिली विधियाँ
स्थिरांक
सार्वजनिक स्थैतिक अंतिम स्ट्रिंग OP_NAME
इस ऑप का नाम, जैसा कि TensorFlow कोर इंजन द्वारा जाना जाता है
स्थिर मान: "TPURepliateMetadata"
सार्वजनिक तरीके
सार्वजनिक स्थैतिक रेप्लिकेटमेटाडेटा.ऑप्शंस कंप्यूटेशनशेप (सूची<लंबा> कंप्यूटेशनशेप)
पैरामीटर
अभिकलनआकार | पदावनत. इसके बजाय num_cores_per_replika का उपयोग करें। |
---|
सार्वजनिक स्थैतिक रेप्लिकेटमेटाडेटा बनाएं ( स्कोप स्कोप, लंबी संख्या प्रतिकृतियां, विकल्प... विकल्प)
एक नया रेप्लिकेटमेटाडेटा ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|---|
संख्याप्रतिकृतियाँ | गणना की प्रतिकृतियों की संख्या |
विकल्प | वैकल्पिक गुण मान रखता है |
रिटर्न
- रिप्लिकेटमेटाडेटा का एक नया उदाहरण
सार्वजनिक स्थैतिक रेप्लिकेटमेटाडेटा.ऑप्शंस डिवाइसअसाइनमेंट (सूची<लंबा> डिवाइसअसाइनमेंट)
पैरामीटर
डिवाइस असाइनमेंट | गणना के लिए उपकरणों का असाइनमेंट. |
---|
सार्वजनिक स्थैतिक रेप्लिकेटमेटाडेटा.ऑप्शंस numCoresPerReplika (लंबे numCoresPerReplusa)
पैरामीटर
numCoresPerReplika | प्रति प्रतिकृति कोर की संख्या. मॉडल समानता के लिए उपयोग किया जाता है। |
---|
सार्वजनिक स्थैतिक ReplicationMetadata.Options टोपोलॉजी (स्ट्रिंग टोपोलॉजी)
पैरामीटर
टोपोलॉजी | टोपोलॉजीप्रोटो टीपीयू पॉड स्लाइस की टोपोलॉजी को दर्शाता है। |
---|
सार्वजनिक स्थैतिक रेप्लिकेटमेटाडेटा.ऑप्शंस उपयोगSpmdForXlaPartitioning (बूलियन उपयोगSpmdForXlaPartitioning)
सार्वजनिक स्थैतिक रेप्लिकेटमेटाडेटा.ऑप्शंस यूज़ टीपीयू (बूलियन यूज़ टीपीयू)
पैरामीटर
टीपीयू का उपयोग करें | क्या गणना को टीपीयू पर रखना है। |
---|