本文整理汇总了Python中tensorflow.python.distribute.cluster_resolver.TPUClusterResolver.task_index方法的典型用法代码示例。如果您正苦于以下问题：Python TPUClusterResolver.task_index方法的具体用法？Python TPUClusterResolver.task_index怎么用？Python TPUClusterResolver.task_index使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.python.distribute.cluster_resolver.TPUClusterResolver的用法示例。
在下文中一共展示了TPUClusterResolver.task_index方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: testOverrideTaskTypeAndIndexAndGetMaster
# 需要导入模块: from tensorflow.python.distribute.cluster_resolver import TPUClusterResolver [as 别名]
# 或者: from tensorflow.python.distribute.cluster_resolver.TPUClusterResolver import task_index [as 别名]
def testOverrideTaskTypeAndIndexAndGetMaster(self):
tpu_map = {
'projects/test-project/locations/us-central1-c/nodes/test-tpu-1': {
'health':
'HEALTHY',
'networkEndpoints': [
{
'ipAddress': '10.2.3.4',
'port': 8470,
},
{
'ipAddress': '10.2.3.5',
'port': 8470,
},
{
'ipAddress': '10.2.3.6',
'port': 8470,
},
{
'ipAddress': '10.2.3.7',
'port': 8470,
},
]
}
}
tpu_cluster_resolver = TPUClusterResolver(
project='test-project',
zone='us-central1-c',
tpu='test-tpu-1',
coordinator_name=None,
credentials=None,
service=self.mock_service_client(tpu_map=tpu_map))
self.assertEqual(tpu_cluster_resolver.master(), 'grpc://10.2.3.4:8470')
tpu_cluster_resolver.task_type = 'worker'
tpu_cluster_resolver.task_index = 3
self.assertEqual(tpu_cluster_resolver.master(), 'grpc://10.2.3.7:8470')
self.assertEqual(
tpu_cluster_resolver.master(
task_type='worker', task_index=2, rpc_layer='test'),
'test://10.2.3.6:8470')