Searched defs:device_coordinates (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | topology.py | 52 def __init__(self, serialized=None, mesh_shape=None, device_coordinates=None): argument
160 def device_coordinates(self): member in Topology
178 def task_ordinal_at_coordinates(self, device_coordinates): argument
191 def tpu_device_ordinal_at_coordinates(self, device_coordinates): argument
204 def cpu_device_name_at_coordinates(self, device_coordinates, job=None): argument
209 def tpu_device_name_at_coordinates(self, device_coordinates, job=None): argument
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | topology.proto | 26 repeated int32 device_coordinates = 4; field
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util_test.cc | 131 llvm::ArrayRef<int> device_coordinates) { in TopologyWithDeviceCoordinates()
610 auto device_coordinates = status_or_device_coodinates.ConsumeValueOrDie(); in TEST() local
|
| D | tpu_rewrite_device_util.cc | 430 llvm::SmallVector<int64_t, 8> device_coordinates; in GetDeviceCoordinates() local
|