Searched refs:z_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 225 z_size: int) -> List[Tuple[int, int, int]]:
279 return [(0, 0, i) for i in range(z_size)]
280 if x_size == 1 and z_size == 1:
282 if y_size == 1 and z_size == 1:
288 y_size % 2 != 0) or (z_size > 1 and
289 z_size % 2 != 0):
292 for z in range(z_size):
301 if z_size == 1:
306 ret.extend((x, y, z) for (x, z, y) in _open_ring_2d(x_size, z_size, 0))
310 ret.extend((x, y, z) for (y, z, x) in _open_ring_2d(y_size, z_size, 0))
[all …]
|
| /external/OpenCL-CTS/test_conformance/thread_dimensions/ |
| D | test_thread_dimensions.cpp | 560 cl_uint x_size, y_size, z_size; in test_thread_dimensions() local
563 z_size = min_z_size; in test_thread_dimensions()
564 while (z_size <= max_z_size) { in test_thread_dimensions()
570 … log_info("Base test size %s:\n", print_dimensions(x_size, y_size, z_size, dimensions)); in test_thread_dimensions()
577 final_z_size = z_size; in test_thread_dimensions()
626 …final_z_size = (int)get_random_float(0, (z_size/size_increase_per_iteration), d)+z_size/size_incre… in test_thread_dimensions()
802 if (z_size == max_z_size) in test_thread_dimensions()
804 z_size *= size_increase_per_iteration; in test_thread_dimensions()
805 if (z_size > max_z_size) in test_thread_dimensions()
806 z_size = max_z_size; in test_thread_dimensions()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | work_group_picking.cc | 191 int z_size = std::max(1, 32 / (x_size * y_size)); in GetWorkGroupSizeForApple() local
192 z_size = std::min(z_size, static_cast<int>(grid_size.z)); in GetWorkGroupSizeForApple()
193 return {x_size, y_size, z_size}; in GetWorkGroupSizeForApple()
|