TPUReshardVariables

ओप जो ऑन-डिवाइस टीपीयू वेरिएबल्स को निर्दिष्ट स्थिति में पुनः लोड करता है।

ओप जो ऑन-डिवाइस टीपीयू वेरिएबल्स को निर्दिष्ट स्थिति में पुनः लोड करता है। केवल आंतरिक प्रयोग हेतु।

शार्डिंग स्थिति को संकलन की कुंजी के रूप में दर्शाया गया है जिसने मुख्य प्रोग्राम के साथ शार्डिंग/अनशार्डिंग प्रोग्राम उत्पन्न किया है। new_format_key वांछित स्थिति निर्दिष्ट करता है, और format_state_var चर की वर्तमान स्थिति है।