| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv2d_acl.cc | 32 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in ACLDepthwiseConvImpl()
130 int64_t kernel_rows, int64_t kernel_cols, in ACLGemmConvImpl()
236 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_ACLConv2DF32()
|
| H A D | runtime_single_threaded_conv2d.cc | 25 int64_t input_cols, int64_t input_channels, int64_t kernel_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
45 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
|
| H A D | runtime_conv2d.cc | 28 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenConv2DF32()
50 int64_t input_cols, int64_t input_channels, int64_t kernel_rows, in __xla_cpu_runtime_EigenConv2DF16()
|
| H A D | runtime_conv2d_mkl.cc | 55 int64_t kernel_rows, int64_t kernel_cols, in MKLConvImpl()
157 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_MKLConv2DF32()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/ |
| H A D | 8b_mla.cpp | 30 const unsigned int kernel_rows, in generic_get_packed_size()
42 const unsigned int kernel_rows, in generic_pack()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/ |
| H A D | addressing.hpp | 191 unsigned int kernel_rows, unsigned int kernel_cols, in fill_pointer_array_generic_kernel()
240 unsigned int kernel_rows, unsigned int kernel_cols, in fill_patch_array_generic_kernel()
|
| H A D | addressing.cpp | 92 const unsigned int kernel_rows, const unsigned int kernel_cols, in fill_pointer_array_generic_kernel()
263 const unsigned int kernel_rows, const unsigned int kernel_cols, in fill_patch_array_generic_kernel()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/ |
| H A D | CpuDepthwiseConv2dAssemblyWrapperKernel.cpp | 75 const unsigned int kernel_rows = weights->dimension(idx_height); in create_arm_dwc() local
116 const unsigned int kernel_rows = weights->dimension(idx_height); in create_arm_dwc_quant() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/ |
| H A D | output_transform.hpp | 151 unsigned int kernel_rows, unsigned int kernel_cols) in TransformBase()
273 unsigned int kernel_rows, unsigned int kernel_cols, in TransformUnpadded()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | nn_ops_test.cc | 853 int rows, int cols, int depth, int kernel_rows, in BM_AvgPool()
943 int rows, int cols, int depth, int kernel_rows, in BM_AvgPoolBk()
1045 int rows, int cols, int depth, int kernel_rows, in BM_MaxPool()
1139 int rows, int cols, int depth, int kernel_rows, in BM_MaxPoolBk()
|
| /aosp_15_r20/external/ruy/ruy/ |
| H A D | block_map.cc | 172 void GetRectangularness(int rows, int cols, int kernel_rows, int kernel_cols, in GetRectangularness()
339 void MakeBlockMap(int rows, int cols, int depth, int kernel_rows, in MakeBlockMap()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/ |
| H A D | sme2_s8q_planar_3x3_s1_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_s8q_planar_3x3_s1_4rows_dot_za
|
| H A D | sme2_s8q_planar_3x3_s2_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_s8q_planar_3x3_s2_4rows_dot_za
|
| H A D | sme2_s8q_planar_5x5_s2_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_s8q_planar_5x5_s2_4rows_dot_za
|
| H A D | sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za
|
| H A D | sme2_fp32_planar_5x5_s2_4rows_mla_za.hpp | 56 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_fp32_planar_5x5_s2_4rows_mla_za
|
| H A D | sme2_s8q_planar_5x5_s1_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_s8q_planar_5x5_s1_4rows_dot_za
|
| H A D | sme2_fp32_planar_3x3_s2_4rows_mla_za.hpp | 56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32_planar_3x3_s2_4rows_mla_za
|
| H A D | sme2_fp32bf16fp32_planar_3x3_s2_4rows_dot_za.hpp | 56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_3x3_s2_4rows_dot_za
|
| H A D | sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za
|
| H A D | sme2_u8q_planar_5x5_s2_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8q_planar_5x5_s2_4rows_dot_za
|
| H A D | sme2_u8q_planar_5x5_s1_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8q_planar_5x5_s1_4rows_dot_za
|
| H A D | sme2_fp32_planar_3x3_s1_4rows_mla_za.hpp | 56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32_planar_3x3_s1_4rows_mla_za
|
| H A D | sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za.hpp | 54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za
|
| H A D | sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za.hpp | 56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za
|