| /external/pthreadpool/include/ |
| D | pthreadpool.h | 1581 size_t tile_m) in call_wrapper_5d_tile_1d()
1589 size_t tile_l, size_t tile_m) in call_wrapper_5d_tile_2d()
1610 size_t tile_m, size_t tile_n) in call_wrapper_6d_tile_2d()
|
| /external/pthreadpool/test/ |
| D | pthreadpool-cxx.cc | 1187 [](size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in TEST()
1205 [](size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in TEST()
1223 [&indicators](size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in TEST()
1255 [&counters](size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in TEST()
1297 [](size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in TEST()
1316 [](size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in TEST()
1339 …tors](size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in TEST()
1373 …ters](size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in TEST()
1619 …[](size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t start_n, size_t tile_m, size_t t… in TEST()
1639 …[](size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t start_n, size_t tile_m, size_t t… in TEST()
[all …]
|
| D | pthreadpool.cc | 8078 …CheckBounds5DTile1D(void*, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in CheckBounds5DTile1D()
8117 …CheckTiling5DTile1D(void*, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in CheckTiling5DTile1D()
8154 …ool* processed_indicators, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in SetTrue5DTile1D()
8223 …c_int* processed_counters, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in Increment5DTile1D()
8364 …_int* num_processed_items, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in IncrementSame5DTile1D()
8390 …_int* num_processed_items, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in WorkImbalance5DTile1D()
8452 …oid*, size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in CheckBounds5DTile2D()
8492 …oid*, size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in CheckTiling5DTile2D()
8534 …tors, size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in SetTrue5DTile2D()
8605 …ters, size_t i, size_t j, size_t k, size_t start_l, size_t start_m, size_t tile_l, size_t tile_m) { in Increment5DTile2D()
[all …]
|
| /external/pthreadpool/src/ |
| D | threadpool-object.h | 478 size_t tile_m; member
513 size_t tile_m; member
606 size_t tile_m; member
|
| D | shim.c | 467 size_t tile_m, in pthreadpool_parallelize_5d_tile_1d()
493 size_t tile_m, in pthreadpool_parallelize_5d_tile_2d()
575 size_t tile_m, in pthreadpool_parallelize_6d_tile_2d()
|
| D | portable-api.c | 1239 const size_t tile_m = threadpool->params.parallelize_5d_tile_1d.tile_m; in thread_parallelize_5d_tile_1d() local
1308 const size_t tile_m = threadpool->params.parallelize_5d_tile_2d.tile_m; in thread_parallelize_5d_tile_2d() local
1527 const size_t tile_m = threadpool->params.parallelize_6d_tile_2d.tile_m; in thread_parallelize_6d_tile_2d() local
2806 size_t tile_m, in pthreadpool_parallelize_5d_tile_1d()
2867 size_t tile_m, in pthreadpool_parallelize_5d_tile_2d()
3054 size_t tile_m, in pthreadpool_parallelize_6d_tile_2d()
|
| D | fastpath.c | 1353 const size_t tile_m = threadpool->params.parallelize_5d_tile_1d.tile_m; in pthreadpool_thread_parallelize_5d_tile_1d_fastpath() local
1427 const size_t tile_m = threadpool->params.parallelize_5d_tile_2d.tile_m; in pthreadpool_thread_parallelize_5d_tile_2d_fastpath() local
1661 const size_t tile_m = threadpool->params.parallelize_6d_tile_2d.tile_m; in pthreadpool_thread_parallelize_6d_tile_2d_fastpath() local
|
| /external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/ |
| D | mma_accum_lambda_iterator.h | 164 for (int tile_m = 0; tile_m < Policy::TileIterations::kRow; ++tile_m) { in iterateRows() local
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 101 size_t tile_m) in xnn_compute_transposec_5d()
128 size_t tile_m, in xnn_compute_transposec_6d()
241 size_t tile_m) in xnn_compute_transposev_5d()
271 size_t tile_m, in xnn_compute_transposev_6d()
|