TPUReshardVariables

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

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

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