Searched defs:bound_z (Results 1 – 1 of 1) sorted by relevance
225 int bound_y, int bound_z, int bound_core) { in DeviceCoordinateOutOfBound()233 int bound_z, int bound_core) { in DeviceCoordinateErrorMsg()303 const int bound_z = topology_proto.mesh_shape(2); in ParseTopologyAttr() local365 const int bound_z = topology.n3(); in GetGeneralTPUExecutionDeviceAssignment() local