Home
last modified time | relevance | path

Searched defs:tile_j (Results 1 – 25 of 35) sorted by relevance

12

/aosp_15_r20/external/pytorch/caffe2/utils/threadpool/
H A Dpthreadpool.cc121 size_t tile_j; member
135 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_2d_tiled() local
146 size_t tile_j) in legacy_pthreadpool_compute_2d_tiled()
183 size_t tile_j; member
203 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_3d_tiled() local
217 size_t tile_j, in legacy_pthreadpool_compute_3d_tiled()
274 size_t tile_j; member
299 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_4d_tiled() local
323 size_t tile_j, in legacy_pthreadpool_compute_4d_tiled()
/aosp_15_r20/external/pthreadpool/src/
H A Dlegacy-api.c56 size_t tile_j) in pthreadpool_compute_2d_tiled()
72 size_t tile_j; member
88 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_3d_tiled() local
101 size_t tile_j, in pthreadpool_compute_3d_tiled()
148 size_t tile_j; member
169 const size_t tile_j = min(max_tile_j, context->range_j - index_j); in compute_4d_tiled() local
184 size_t tile_j, in pthreadpool_compute_4d_tiled()
H A Dshim.c89 size_t tile_j, in pthreadpool_parallelize_2d_tile_1d()
106 size_t tile_j, in pthreadpool_parallelize_2d_tile_2d()
125 size_t tile_j, in pthreadpool_parallelize_2d_tile_2d_with_uarch()
180 size_t tile_j, in pthreadpool_parallelize_3d_tile_2d()
203 size_t tile_j, in pthreadpool_parallelize_3d_tile_2d_with_uarch()
H A Dthreadpool-object.h121 size_t tile_j; member
144 size_t tile_j; member
175 size_t tile_j; member
220 size_t tile_j; member
255 size_t tile_j; member
H A Dportable-api.c192 const size_t tile_j = threadpool->params.parallelize_2d_tile_1d.tile_j; in thread_parallelize_2d_tile_1d() local
238 const size_t tile_j = threadpool->params.parallelize_2d_tile_2d.tile_j; in thread_parallelize_2d_tile_2d() local
297 const size_t tile_j = threadpool->params.parallelize_2d_tile_2d_with_uarch.tile_j; in thread_parallelize_2d_tile_2d_with_uarch() local
444 const size_t tile_j = threadpool->params.parallelize_3d_tile_2d.tile_j; in thread_parallelize_3d_tile_2d() local
509 const size_t tile_j = threadpool->params.parallelize_3d_tile_2d_with_uarch.tile_j; in thread_parallelize_3d_tile_2d_with_uarch() local
1405 size_t tile_j, in pthreadpool_parallelize_2d_tile_1d()
1452 size_t tile_j, in pthreadpool_parallelize_2d_tile_2d()
1504 size_t tile_j, in pthreadpool_parallelize_2d_tile_2d_with_uarch()
1661 size_t tile_j, in pthreadpool_parallelize_3d_tile_2d()
1717 size_t tile_j, in pthreadpool_parallelize_3d_tile_2d_with_uarch()
H A Dfastpath.c211 const size_t tile_j = threadpool->params.parallelize_2d_tile_1d.tile_j; in pthreadpool_thread_parallelize_2d_tile_1d_fastpath() local
262 const size_t tile_j = threadpool->params.parallelize_2d_tile_2d.tile_j; in pthreadpool_thread_parallelize_2d_tile_2d_fastpath() local
326 const size_t tile_j = threadpool->params.parallelize_2d_tile_2d_with_uarch.tile_j; in pthreadpool_thread_parallelize_2d_tile_2d_with_uarch_fastpath() local
488 const size_t tile_j = threadpool->params.parallelize_3d_tile_2d.tile_j; in pthreadpool_thread_parallelize_3d_tile_2d_fastpath() local
558 const size_t tile_j = threadpool->params.parallelize_3d_tile_2d_with_uarch.tile_j; in pthreadpool_thread_parallelize_3d_tile_2d_with_uarch_fastpath() local
/aosp_15_r20/external/pthreadpool/test/
H A Dpthreadpool.cc1292 static void CheckBounds2DTile1D(void*, size_t i, size_t start_j, size_t tile_j) { in CheckBounds2DTile1D()
1326 static void CheckTiling2DTile1D(void*, size_t i, size_t start_j, size_t tile_j) { in CheckTiling2DTile1D()
1361 … SetTrue2DTile1D(std::atomic_bool* processed_indicators, size_t i, size_t start_j, size_t tile_j) { in SetTrue2DTile1D()
1416 …d Increment2DTile1D(std::atomic_int* processed_counters, size_t i, size_t start_j, size_t tile_j) { in Increment2DTile1D()
1529 …rementSame2DTile1D(std::atomic_int* num_processed_items, size_t i, size_t start_j, size_t tile_j) { in IncrementSame2DTile1D()
1554 …kImbalance2DTile1D(std::atomic_int* num_processed_items, size_t i, size_t start_j, size_t tile_j) { in WorkImbalance2DTile1D()
1615 static void CheckBounds2DTile2D(void*, size_t start_i, size_t start_j, size_t tile_i, size_t tile_j… in CheckBounds2DTile2D()
1652 static void CheckTiling2DTile2D(void*, size_t start_i, size_t start_j, size_t tile_i, size_t tile_j… in CheckTiling2DTile2D()
1694 …:atomic_bool* processed_indicators, size_t start_i, size_t start_j, size_t tile_i, size_t tile_j) { in SetTrue2DTile2D()
1753 …td::atomic_int* processed_counters, size_t start_i, size_t start_j, size_t tile_i, size_t tile_j) { in Increment2DTile2D()
[all …]
/aosp_15_r20/external/XNNPACK/src/
H A Doperator-run.c29 size_t tile_j) in xnn_compute_transposec_2d()
47 size_t tile_j, in xnn_compute_transposec_3d()
155 size_t tile_j) in xnn_compute_transposev_2d()
181 size_t tile_j, in xnn_compute_transposev_3d()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sme2_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sme2_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sme2_fp32_nhwc_3x3_s1_output4x4_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst_direct_impl() member

12