Searched refs:bound_x (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 224 bool DeviceCoordinateOutOfBound(int x, int y, int z, int core, int bound_x, in DeviceCoordinateOutOfBound() argument
226 return x < 0 || x >= bound_x || y < 0 || y >= bound_y || z < 0 || in DeviceCoordinateOutOfBound()
232 int z, int core, int bound_x, int bound_y, in DeviceCoordinateErrorMsg() argument
236 "' is outside of mesh shape (", bound_x, ", ", in DeviceCoordinateErrorMsg()
301 const int bound_x = topology_proto.mesh_shape(0); in ParseTopologyAttr() local
306 xla::Array4D<TaskAndDevice> topology(bound_x, bound_y, bound_z, bound_core); in ParseTopologyAttr()
314 if (DeviceCoordinateOutOfBound(x, y, z, core, bound_x, bound_y, bound_z, in ParseTopologyAttr()
316 return DeviceCoordinateErrorMsg(kTopologyAttr, x, y, z, core, bound_x, in ParseTopologyAttr()
363 const int bound_x = topology.n1(); in GetGeneralTPUExecutionDeviceAssignment() local
371 return (x + bound_x * (y + bound_y * z)) * bound_core + core; in GetGeneralTPUExecutionDeviceAssignment()
[all …]
|