Searched refs:bound_y (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 229 int bound_y, int bound_z, int bound_core) { in DeviceCoordinateOutOfBound() argument
230 return x < 0 || x >= bound_x || y < 0 || y >= bound_y || z < 0 || in DeviceCoordinateOutOfBound()
236 int z, int core, int bound_x, int bound_y, in DeviceCoordinateErrorMsg() argument
241 bound_y, ", ", bound_z, ", ", bound_core, ")"); in DeviceCoordinateErrorMsg()
306 const int bound_y = topology_proto.mesh_shape(1); in ParseTopologyAttr() local
310 xla::Array4D<TaskAndDevice> topology(bound_x, bound_y, bound_z, bound_core); in ParseTopologyAttr()
318 if (DeviceCoordinateOutOfBound(x, y, z, core, bound_x, bound_y, bound_z, in ParseTopologyAttr()
321 bound_y, bound_z, bound_core); in ParseTopologyAttr()
368 const int bound_y = topology.n2(); in GetGeneralTPUExecutionDeviceAssignment() local
375 return (x + bound_x * (y + bound_y * z)) * bound_core + core; in GetGeneralTPUExecutionDeviceAssignment()
[all …]
|