सार्वजनिक अंतिम श्रेणी TPURप्रतिकृतिमेटाडेटा
मेटाडेटा इंगित करता है कि टीपीयू गणना को कैसे दोहराया जाना चाहिए।
यह ऑपरेशन मेटाडेटा को एक `tpu.replicate ()` गणना सबग्राफ के संचालन के लिए सामान्य रखता है।
नेस्टेड कक्षाएं
| कक्षा | TPUReplicateMetadata.Options | TPUReplicateMetadata के लिए वैकल्पिक विशेषताएँ | |
सार्वजनिक तरीके
| स्थिर TPUReplicateMetadata.Options | अनुमति दें सॉफ्ट प्लेसमेंट (बूलियन अनुमति सॉफ्ट प्लेसमेंट) |
| स्थिर TPUReplicateMetadata.Options | गणना आकार (सूची <लंबी> गणना आकार) |
| स्थिर TPURप्रतिकृतिमेटाडेटा | create ( स्कोप स्कोप, लॉन्ग numReplicas, Options... विकल्प) एक नया TPUReplicateMetadata ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि। |
| स्थिर TPUReplicateMetadata.Options | डिवाइसअसाइनमेंट (सूची<लॉन्ग> डिवाइसअसाइनमेंट) |
| स्थिर TPUReplicateMetadata.Options | hostComputeCore (सूची <स्ट्रिंग> hostComputeCore) |
| स्थिर TPUReplicateMetadata.Options | numCoresPerReplica (लंबी संख्याCoresPerReplica) |
| स्थिर TPUReplicateMetadata.Options | पैडिंग मैप (सूची <स्ट्रिंग> पैडिंग मैप) |
| स्थिर TPUReplicateMetadata.Options | स्टेपमार्करलोकेशन (स्ट्रिंग स्टेपमार्करलोकेशन) |
| स्थिर TPUReplicateMetadata.Options | टोपोलॉजी (स्ट्रिंग टोपोलॉजी) |
| स्थिर TPUReplicateMetadata.Options | tpuCompileOptionsProto (स्ट्रिंग tpuCompileOptionsProto) |
| स्थिर TPUReplicateMetadata.Options | उपयोगSpmdForXlaPartitioning (बूलियन उपयोगSpmdForXlaPartitioning) |
| स्थिर TPUReplicateMetadata.Options | यूज टीपीयू (बूलियन यूज टीपीयू) |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options अभिकलन आकार (सूची<लंबी> संगणना आकार)
मापदंडों
| संगणनाआकार | पदावनत। इसके बजाय num_cores_per_replica का उपयोग करें। |
|---|
सार्वजनिक स्थैतिक TPUReplicateMetadata बनाएँ ( स्कोप स्कोप, लॉन्ग numReplicas, विकल्प ... विकल्प)
एक नया TPUReplicateMetadata ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
| दायरा | वर्तमान दायरा |
|---|---|
| संख्याप्रतिकृति | गणना की प्रतिकृतियों की संख्या |
| विकल्प | वैकल्पिक विशेषता मान रखता है |
रिटर्न
- TPUReplicateMetadata का एक नया उदाहरण
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options deviceAssignment (सूची<लॉन्ग> डिवाइसअसाइनमेंट)
मापदंडों
| डिवाइस असाइनमेंट | गणना के लिए उपकरणों का असाइनमेंट। |
|---|
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options numCoresPerReplica (लंबी numCoresPerReplica)
मापदंडों
| numCoresPerReplica | प्रति प्रतिकृति कोर की संख्या। मॉडल समांतरता के लिए प्रयुक्त। |
|---|
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options टोपोलॉजी (स्ट्रिंग टोपोलॉजी)
मापदंडों
| टोपोलॉजी | टोपोलॉजीप्रो टीपीयू पॉड स्लाइस की टोपोलॉजी को दर्शाता है। |
|---|
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options tpuCompileOptionsProto (स्ट्रिंग tpuCompileOptionsProto)
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options उपयोगSpmdForXlaPartitioning (बूलियन उपयोगSpmdForXlaPartitioning)
सार्वजनिक स्थैतिक TPUReplicateMetadata.Options useTpu (बूलियन useTpu)
मापदंडों
| टीपीयू का उपयोग करें | क्या गणना को टीपीयू पर रखना है। |
|---|