टीपीयू कोर पर राउंड-रॉबिन लोड संतुलन।
एक लोड संतुलन ऑपशन जो टीपीयू कोर के बीच राउंड-रॉबिन करता है।
यह ऑप [0, NumTPUCoresVisiblePerHost] में पूर्णांकों के बीच राउंड-रॉबिन करता है। यह TensorFlow ऑप्स के साथ इंटरफेसिंग के लिए उपयोगी है जो इनपुट के रूप में एक TPU कोर लेता है जिस पर गणना निष्पादित की जाती है, जैसे `TPUPartitionedCall`।
डिवाइस_ऑर्डिनल: [0, NumTPUCoresVisiblePerHost] में एक पूर्णांक।
सार्वजनिक तरीके
आउटपुट <पूर्णांक> | आउटपुट के रूप में () टेंसर का प्रतीकात्मक हैंडल लौटाता है। |
स्थिर TPURoundRobin | |
आउटपुट <पूर्णांक> |
विरासत में मिली विधियाँ
सार्वजनिक तरीके
सार्वजनिक आउटपुट <पूर्णांक> आउटपुट के रूप में ()
टेंसर का प्रतीकात्मक हैंडल लौटाता है।
TensorFlow संचालन के इनपुट किसी अन्य TensorFlow ऑपरेशन के आउटपुट हैं। इस पद्धति का उपयोग एक प्रतीकात्मक हैंडल प्राप्त करने के लिए किया जाता है जो इनपुट की गणना का प्रतिनिधित्व करता है।
सार्वजनिक स्थैतिक TPURoundRobin निर्माण ( स्कोप स्कोप)
एक नया TPURoundRobin ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
पैरामीटर
दायरा | वर्तमान दायरा |
---|
रिटर्न
- TPURoundRobin का एक नया उदाहरण