Searched refs:bound_z (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
231 z >= bound_z || core < 0 || core >= bound_core; in DeviceCoordinateOutOfBound()
237 int bound_z, int bound_core) { in DeviceCoordinateErrorMsg() argument
241 bound_y, ", ", bound_z, ", ", bound_core, ")"); in DeviceCoordinateErrorMsg()
307 const int bound_z = topology_proto.mesh_shape(2); 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()
369 const int bound_z = topology.n3(); in GetGeneralTPUExecutionDeviceAssignment() local
378 std::vector<bool> used_device_ids(bound_x * bound_y * bound_z * bound_core, in GetGeneralTPUExecutionDeviceAssignment()
[all …]
|