Home
last modified time | relevance | path

Searched defs:tile_i (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/pytorch/caffe2/utils/threadpool/
H A Dpthreadpool.cc120 size_t tile_i; member
134 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_2d_tiled() local
145 size_t tile_i, in legacy_pthreadpool_compute_2d_tiled()
182 size_t tile_i; member
202 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_3d_tiled() local
216 size_t tile_i, in legacy_pthreadpool_compute_3d_tiled()
273 size_t tile_i; member
298 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_4d_tiled() local
322 size_t tile_i, in legacy_pthreadpool_compute_4d_tiled()
/aosp_15_r20/external/pthreadpool/src/
H A Dlegacy-api.c55 size_t tile_i, in pthreadpool_compute_2d_tiled()
71 size_t tile_i; member
87 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_3d_tiled() local
100 size_t tile_i, in pthreadpool_compute_3d_tiled()
147 size_t tile_i; member
168 const size_t tile_i = min(max_tile_i, context->range_i - index_i); in compute_4d_tiled() local
183 size_t tile_i, in pthreadpool_compute_4d_tiled()
/aosp_15_r20/external/pthreadpool/test/
H A Dpthreadpool.cc717 static void CheckBounds1DTile1D(void*, size_t start_i, size_t tile_i) { in CheckBounds1DTile1D()
750 static void CheckTiling1DTile1D(void*, size_t start_i, size_t tile_i) { in CheckTiling1DTile1D()
785 static void SetTrue1DTile1D(std::atomic_bool* processed_indicators, size_t start_i, size_t tile_i) { in SetTrue1DTile1D()
833 static void Increment1DTile1D(std::atomic_int* processed_counters, size_t start_i, size_t tile_i) { in Increment1DTile1D()
929 …c void IncrementSame1DTile1D(std::atomic_int* num_processed_items, size_t start_i, size_t tile_i) { in IncrementSame1DTile1D()
954 …c void WorkImbalance1DTile1D(std::atomic_int* num_processed_items, size_t start_i, size_t tile_i) { in WorkImbalance1DTile1D()
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 …d::atomic_bool* processed_indicators, size_t start_i, size_t start_j, size_t tile_i, size_t tile_j… in SetTrue2DTile2D()
1753 …(std::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/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
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp32_nhwc_5x5_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp32_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_output3x3_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in a64_fp32_nhwc_3x3_s1_output3x3_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sve_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst_direct_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst/
H A Dgeneric_direct.cpp60 uint64_t tile_i = 0, tile_j = 0; in sme2_fp32_nhwc_3x3_s2_output2x2_mla_depthfirst_direct_impl() member

12