| /external/pthreadpool/src/ |
| D | legacy-api.c | 73 size_t tile_k; member
89 const size_t tile_k = min(max_tile_k, context->range_k - index_k); in compute_3d_tiled() local
102 size_t tile_k) in pthreadpool_compute_3d_tiled()
149 size_t tile_k; member
170 const size_t tile_k = min(max_tile_k, context->range_k - index_k); in compute_4d_tiled() local
185 size_t tile_k, in pthreadpool_compute_4d_tiled()
|
| D | shim.c | 224 size_t tile_k, in pthreadpool_parallelize_3d_tile_1d()
243 size_t tile_k, in pthreadpool_parallelize_3d_tile_1d_with_thread()
264 size_t tile_k, in pthreadpool_parallelize_3d_tile_1d_with_uarch()
285 size_t tile_k, in pthreadpool_parallelize_3d_tile_1d_with_uarch_with_thread()
305 size_t tile_k, in pthreadpool_parallelize_3d_tile_2d()
328 size_t tile_k, in pthreadpool_parallelize_3d_tile_2d_with_uarch()
392 size_t tile_k, in pthreadpool_parallelize_4d_tile_2d()
418 size_t tile_k, in pthreadpool_parallelize_4d_tile_2d_with_uarch()
|
| D | threadpool-object.h | 227 size_t tile_k; member
254 size_t tile_k; member
281 size_t tile_k; member
316 size_t tile_k; member
381 size_t tile_k; member
420 size_t tile_k; member
|
| D | portable-api.c | 573 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d.tile_k; in thread_parallelize_3d_tile_1d() local
625 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d.tile_k; in thread_parallelize_3d_tile_1d_with_thread() local
686 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d_with_uarch.tile_k; in thread_parallelize_3d_tile_1d_with_uarch() local
748 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d_with_uarch.tile_k; in thread_parallelize_3d_tile_1d_with_uarch_with_thread() local
801 const size_t tile_k = threadpool->params.parallelize_3d_tile_2d.tile_k; in thread_parallelize_3d_tile_2d() local
866 const size_t tile_k = threadpool->params.parallelize_3d_tile_2d_with_uarch.tile_k; in thread_parallelize_3d_tile_2d_with_uarch() local
1039 const size_t tile_k = threadpool->params.parallelize_4d_tile_2d.tile_k; in thread_parallelize_4d_tile_2d() local
1111 const size_t tile_k = threadpool->params.parallelize_4d_tile_2d_with_uarch.tile_k; in thread_parallelize_4d_tile_2d_with_uarch() local
2162 size_t tile_k, in pthreadpool_parallelize_3d_tile_1d()
2212 size_t tile_k, in pthreadpool_parallelize_3d_tile_1d_with_thread()
[all …]
|
| D | fastpath.c | 632 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d.tile_k; in pthreadpool_thread_parallelize_3d_tile_1d_fastpath() local
689 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d.tile_k; in pthreadpool_thread_parallelize_3d_tile_1d_with_thread_fastpath() local
755 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d_with_uarch.tile_k; in pthreadpool_thread_parallelize_3d_tile_1d_with_uarch_fastpath() local
822 const size_t tile_k = threadpool->params.parallelize_3d_tile_1d_with_uarch.tile_k; in pthreadpool_thread_parallelize_3d_tile_1d_with_uarch_with_thread_fastpath() local
880 const size_t tile_k = threadpool->params.parallelize_3d_tile_2d.tile_k; in pthreadpool_thread_parallelize_3d_tile_2d_fastpath() local
950 const size_t tile_k = threadpool->params.parallelize_3d_tile_2d_with_uarch.tile_k; in pthreadpool_thread_parallelize_3d_tile_2d_with_uarch_fastpath() local
1138 const size_t tile_k = threadpool->params.parallelize_4d_tile_2d.tile_k; in pthreadpool_thread_parallelize_4d_tile_2d_fastpath() local
1215 const size_t tile_k = threadpool->params.parallelize_4d_tile_2d_with_uarch.tile_k; in pthreadpool_thread_parallelize_4d_tile_2d_with_uarch_fastpath() local
|
| /external/pytorch/caffe2/utils/threadpool/ |
| D | pthreadpool.cc | 184 size_t tile_k; member
204 const size_t tile_k = min(max_tile_k, context->range_k - index_k); in compute_3d_tiled() local
218 size_t tile_k) { in legacy_pthreadpool_compute_3d_tiled()
275 size_t tile_k; member
300 const size_t tile_k = min(max_tile_k, context->range_k - index_k); in compute_4d_tiled() local
324 size_t tile_k, in legacy_pthreadpool_compute_4d_tiled()
|
| /external/pthreadpool/include/ |
| D | pthreadpool.h | 1539 size_t tile_k) in call_wrapper_3d_tile_1d()
1547 size_t tile_j, size_t tile_k) in call_wrapper_3d_tile_2d()
1568 size_t tile_k, size_t tile_l) in call_wrapper_4d_tile_2d()
|
| /external/pthreadpool/test/ |
| D | pthreadpool.cc | 3989 static void CheckBounds3DTile1D(void*, size_t i, size_t j, size_t start_k, size_t tile_k) { in CheckBounds3DTile1D()
4026 static void CheckTiling3DTile1D(void*, size_t i, size_t j, size_t start_k, size_t tile_k) { in CheckTiling3DTile1D()
4063 …Tile1D(std::atomic_bool* processed_indicators, size_t i, size_t j, size_t start_k, size_t tile_k) { in SetTrue3DTile1D()
4124 …t3DTile1D(std::atomic_int* processed_counters, size_t i, size_t j, size_t start_k, size_t tile_k) { in Increment3DTile1D()
4249 …3DTile1D(std::atomic_int* num_processed_items, size_t i, size_t j, size_t start_k, size_t tile_k) { in IncrementSame3DTile1D()
4275 …3DTile1D(std::atomic_int* num_processed_items, size_t i, size_t j, size_t start_k, size_t tile_k) { in WorkImbalance3DTile1D()
4337 …d CheckBounds3DTile1DWithThread(void*, size_t, size_t i, size_t j, size_t start_k, size_t tile_k) { in CheckBounds3DTile1DWithThread() argument
4374 …d CheckTiling3DTile1DWithThread(void*, size_t, size_t i, size_t j, size_t start_k, size_t tile_k) { in CheckTiling3DTile1DWithThread() argument
4411 …td::atomic_bool* processed_indicators, size_t, size_t i, size_t j, size_t start_k, size_t tile_k) { in SetTrue3DTile1DWithThread() argument
4472 …d(std::atomic_int* processed_counters, size_t, size_t i, size_t j, size_t start_k, size_t tile_k) { in Increment3DTile1DWithThread() argument
[all …]
|
| D | pthreadpool-cxx.cc | 592 [](size_t i, size_t j, size_t start_k, size_t tile_k) { in TEST()
608 [](size_t i, size_t j, size_t start_k, size_t tile_k) { in TEST()
626 [&indicators](size_t i, size_t j, size_t start_k, size_t tile_k) { in TEST()
654 [&counters](size_t i, size_t j, size_t start_k, size_t tile_k) { in TEST()
692 [](size_t i, size_t start_j, size_t start_k, size_t tile_j, size_t tile_k) { in TEST()
709 [](size_t i, size_t start_j, size_t start_k, size_t tile_j, size_t tile_k) { in TEST()
732 [&indicators](size_t i, size_t start_j, size_t start_k, size_t tile_j, size_t tile_k) { in TEST()
762 [&counters](size_t i, size_t start_j, size_t start_k, size_t tile_j, size_t tile_k) { in TEST()
987 [](size_t i, size_t j, size_t start_k, size_t start_l, size_t tile_k, size_t tile_l) { in TEST()
1005 [](size_t i, size_t j, size_t start_k, size_t start_l, size_t tile_k, size_t tile_l) { in TEST()
[all …]
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 48 size_t tile_k) in xnn_compute_transposec_3d()
73 size_t tile_k, in xnn_compute_transposec_4d()
182 size_t tile_k) in xnn_compute_transposev_3d()
210 size_t tile_k, in xnn_compute_transposev_4d()
|
| /external/pytorch/torch/sparse/ |
| D | _triton_ops.py | 1518 tile_k, argument
|