• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1path: "tensorflow.tpu.experimental.DeviceAssignment"
2tf_class {
3  is_instance: "<class \'tensorflow.python.tpu.device_assignment.DeviceAssignment\'>"
4  is_instance: "<type \'object\'>"
5  member {
6    name: "core_assignment"
7    mtype: "<type \'property\'>"
8  }
9  member {
10    name: "num_cores_per_replica"
11    mtype: "<type \'property\'>"
12  }
13  member {
14    name: "num_replicas"
15    mtype: "<type \'property\'>"
16  }
17  member {
18    name: "topology"
19    mtype: "<type \'property\'>"
20  }
21  member_method {
22    name: "__init__"
23    argspec: "args=[\'self\', \'topology\', \'core_assignment\'], varargs=None, keywords=None, defaults=None"
24  }
25  member_method {
26    name: "build"
27    argspec: "args=[\'topology\', \'computation_shape\', \'computation_stride\', \'num_replicas\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'1\'], "
28  }
29  member_method {
30    name: "coordinates"
31    argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=None"
32  }
33  member_method {
34    name: "host_device"
35    argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'None\'], "
36  }
37  member_method {
38    name: "lookup_replicas"
39    argspec: "args=[\'self\', \'task_id\', \'logical_core\'], varargs=None, keywords=None, defaults=None"
40  }
41  member_method {
42    name: "tpu_device"
43    argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'None\'], "
44  }
45  member_method {
46    name: "tpu_ordinal"
47    argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=[\'0\', \'0\'], "
48  }
49}
50