path: "tensorflow.tpu.experimental.DeviceAssignment" tf_class { is_instance: "" is_instance: "" member { name: "core_assignment" mtype: "" } member { name: "num_cores_per_replica" mtype: "" } member { name: "num_replicas" mtype: "" } member { name: "topology" mtype: "" } member_method { name: "__init__" argspec: "args=[\'self\', \'topology\', \'core_assignment\'], varargs=None, keywords=None, defaults=None" } member_method { name: "build" argspec: "args=[\'topology\', \'computation_shape\', \'computation_stride\', \'num_replicas\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'1\'], " } member_method { name: "coordinates" argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=None" } member_method { name: "host_device" argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'None\'], " } member_method { name: "lookup_replicas" argspec: "args=[\'self\', \'task_id\', \'logical_core\'], varargs=None, keywords=None, defaults=None" } member_method { name: "tpu_device" argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'None\'], " } member_method { name: "tpu_ordinal" argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=[\'0\', \'0\'], " } }