Home
last modified time | relevance | path

Searched defs:kernel_rows (Results 1 – 25 of 105) sorted by relevance

12345

/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/
D8b_mla.cpp30 const unsigned int kernel_rows, in generic_get_packed_size()
42 const unsigned int kernel_rows, in generic_pack()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc25 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()
Druntime_conv2d_acl.cc32 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()
Druntime_conv2d.cc28 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()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/
Daddressing.hpp191 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()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/
Dsme2_u8s8u8q_planar_3x3_s2_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za
Dsme2_u8q_planar_3x3_s1_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_u8q_planar_3x3_s1_4rows_dot_za
Dsme2_u8q_planar_3x3_s2_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_u8q_planar_3x3_s2_4rows_dot_za
Dsme2_u8s8u8q_planar_5x5_s1_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za
Dsme2_fp32_planar_3x3_s2_4rows_mla_za.hpp56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32_planar_3x3_s2_4rows_mla_za
Dsme2_u8s8u8q_planar_5x5_s2_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za
Dsme2_s8q_planar_5x5_s2_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_s8q_planar_5x5_s2_4rows_dot_za
Dsme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za.hpp56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za
Dsme2_s8q_planar_5x5_s1_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_s8q_planar_5x5_s1_4rows_dot_za
Dsme2_s8q_planar_3x3_s2_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_s8q_planar_3x3_s2_4rows_dot_za
Dsme2_u8s8u8q_planar_3x3_s1_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za
Dsme2_fp32bf16fp32_planar_5x5_s2_4rows_dot_za.hpp56 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_5x5_s2_4rows_dot_za
Dsme2_fp32_planar_3x3_s1_4rows_mla_za.hpp56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32_planar_3x3_s1_4rows_mla_za
Dsme2_fp32_planar_5x5_s1_4rows_mla_za.hpp56 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_fp32_planar_5x5_s1_4rows_mla_za
Dsme2_u8q_planar_5x5_s1_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8q_planar_5x5_s1_4rows_dot_za
Dsme2_fp32bf16fp32_planar_5x5_s1_4rows_dot_za.hpp56 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_5x5_s1_4rows_dot_za
Dsme2_s8q_planar_3x3_s1_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_s8q_planar_3x3_s1_4rows_dot_za
Dsme2_fp32_planar_5x5_s2_4rows_mla_za.hpp56 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_fp32_planar_5x5_s2_4rows_mla_za
Dsme2_u8q_planar_5x5_s2_4rows_dot_za.hpp54 constexpr static auto kernel_rows = 5u, kernel_cols = 5u; member in arm_conv::depthwise::sme2_u8q_planar_5x5_s2_4rows_dot_za
Dsme2_fp32bf16fp32_planar_3x3_s2_4rows_dot_za.hpp56 constexpr static auto kernel_rows = 3u, kernel_cols = 3u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_3x3_s2_4rows_dot_za

12345