1path: "tensorflow.tpu.experimental" 2tf_module { 3 member { 4 name: "DeviceAssignment" 5 mtype: "<type \'type\'>" 6 } 7 member { 8 name: "TPUSystemMetadata" 9 mtype: "<type \'type\'>" 10 } 11 member { 12 name: "Topology" 13 mtype: "<type \'type\'>" 14 } 15 member { 16 name: "embedding" 17 mtype: "<type \'module\'>" 18 } 19 member_method { 20 name: "initialize_tpu_system" 21 argspec: "args=[\'cluster_resolver\'], varargs=None, keywords=None, defaults=[\'None\'], " 22 } 23 member_method { 24 name: "shutdown_tpu_system" 25 argspec: "args=[\'cluster_resolver\'], varargs=None, keywords=None, defaults=[\'None\'], " 26 } 27} 28