Searched defs:topology_proto (Results 1 – 4 of 4) sorted by relevance
116 tpu::TopologyProto topology_proto; in TopologyWithMeshShape() local124 tpu::TopologyProto topology_proto; in TopologyWithMeshShapeAndTasks() local133 tpu::TopologyProto topology_proto; in TopologyWithDeviceCoordinates() local285 tpu::TopologyProto topology_proto; in TEST() local376 tpu::TopologyProto topology_proto; in TEST() local498 tpu::TopologyProto topology_proto; in TEST() local
262 tpu::TopologyProto topology_proto; in ParseTopologyAttr() local
194 def _set_global_tpu_array_fn(topology_proto): argument
83 tpu::TopologyProto topology_proto; in GetTPUTopology() local