| /aosp_15_r20/external/pthreadpool/src/ |
| H A D | shim.c | 343 size_t tile_m, in pthreadpool_parallelize_5d_tile_1d()
369 size_t tile_m, in pthreadpool_parallelize_5d_tile_2d()
451 size_t tile_m, in pthreadpool_parallelize_6d_tile_2d()
|
| H A D | threadpool-object.h | 425 size_t tile_m; member
460 size_t tile_m; member
553 size_t tile_m; member
|
| H A D | portable-api.c | 883 const size_t tile_m = threadpool->params.parallelize_5d_tile_1d.tile_m; in thread_parallelize_5d_tile_1d() local
952 const size_t tile_m = threadpool->params.parallelize_5d_tile_2d.tile_m; in thread_parallelize_5d_tile_2d() local
1171 const size_t tile_m = threadpool->params.parallelize_6d_tile_2d.tile_m; in thread_parallelize_6d_tile_2d() local
2079 size_t tile_m, in pthreadpool_parallelize_5d_tile_1d()
2140 size_t tile_m, in pthreadpool_parallelize_5d_tile_2d()
2327 size_t tile_m, in pthreadpool_parallelize_6d_tile_2d()
|
| H A D | fastpath.c | 962 const size_t tile_m = threadpool->params.parallelize_5d_tile_1d.tile_m; in pthreadpool_thread_parallelize_5d_tile_1d_fastpath() local
1036 const size_t tile_m = threadpool->params.parallelize_5d_tile_2d.tile_m; in pthreadpool_thread_parallelize_5d_tile_2d_fastpath() local
1270 const size_t tile_m = threadpool->params.parallelize_6d_tile_2d.tile_m; in pthreadpool_thread_parallelize_6d_tile_2d_fastpath() local
|
| /aosp_15_r20/external/pthreadpool/test/ |
| H A D | pthreadpool.cc | 5555 …CheckBounds5DTile1D(void*, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in CheckBounds5DTile1D()
5594 …CheckTiling5DTile1D(void*, size_t i, size_t j, size_t k, size_t l, size_t start_m, size_t tile_m) { in CheckTiling5DTile1D()
5631 …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()
5700 …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()
5841 …_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()
5867 …_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()
5929 …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()
5969 …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()
6011 …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()
6082 …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 …]
|
| /aosp_15_r20/external/XNNPACK/src/ |
| H A 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/ |
| H A D | mma_accum_lambda_iterator.h | 164 for (int tile_m = 0; tile_m < Policy::TileIterations::kRow; ++tile_m) { in iterateRows() local
|