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