| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/ |
| D | addressing.hpp | 89 unsigned int kernel_rows, unsigned int kernel_cols,
142 unsigned int kernel_rows, unsigned int kernel_cols,
191 unsigned int kernel_rows, unsigned int kernel_cols, in fill_pointer_array_generic_kernel() argument
203 kernel_rows, kernel_cols, in fill_pointer_array_generic_kernel()
240 unsigned int kernel_rows, unsigned int kernel_cols, in fill_patch_array_generic_kernel() argument
253 kernel_rows, kernel_cols, in fill_patch_array_generic_kernel()
|
| D | addressing.cpp | 92 const unsigned int kernel_rows, const unsigned int kernel_cols, in fill_pointer_array_generic_kernel() argument
121 for (; ii < pad_top && ki < kernel_rows; ii++, ki++) in fill_pointer_array_generic_kernel()
130 for (; ii < last_valid_row && ki < kernel_rows; ii++, ki++) in fill_pointer_array_generic_kernel()
151 for (; ki < kernel_rows; ki++) in fill_pointer_array_generic_kernel()
263 const unsigned int kernel_rows, const unsigned int kernel_cols, in fill_patch_array_generic_kernel() argument
280 const auto input_rows = kernel_rows + (output_rows - 1) * stride_rows; in fill_patch_array_generic_kernel()
286 for (auto ki = 0u; ki < kernel_rows; ki++) in fill_patch_array_generic_kernel()
|
| /external/ruy/ruy/ |
| D | block_map.cc | 172 void GetRectangularness(int rows, int cols, int kernel_rows, int kernel_cols, in GetRectangularness() argument
193 std::max(0, floor_log2(rows) - pot_log2(kernel_rows) - in GetRectangularness()
199 ceil_log2(rows) - pot_log2(kernel_rows); in GetRectangularness()
339 void MakeBlockMap(int rows, int cols, int depth, int kernel_rows, in MakeBlockMap() argument
346 RUY_DCHECK_GE(rows, kernel_rows); in MakeBlockMap()
348 RUY_DCHECK_EQ(rows % kernel_rows, 0); in MakeBlockMap()
355 GetRectangularness(rows, cols, kernel_rows, kernel_cols, in MakeBlockMap()
358 const int kernel_rows_log2 = pot_log2(kernel_rows); in MakeBlockMap()
438 round_down_pot(rows >> num_blocks_of_rows_log2, kernel_rows); in MakeBlockMap()
443 kernel_rows) >> in MakeBlockMap()
[all …]
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/ |
| D | 8b_mla.cpp | 30 const unsigned int kernel_rows, in generic_get_packed_size() argument
36 …return arm_gemm::roundup((long unsigned int) n_input_channels, per_iter) * kernel_rows * kernel_co… in generic_get_packed_size()
42 const unsigned int kernel_rows, in generic_pack() argument
65 for (unsigned int i = 0; i < kernel_rows; i++) in generic_pack()
|
| D | generic_quantized_dot_product.cpp | 50 4 * n_dots_per_kernel_row * args.kernel_rows * sizeof(int8_t) + // Weights in get_storage_size()
75 4 * n_dots_per_kernel_row * args.kernel_rows * sizeof(T) + // Weights in pack_parameters()
96 buffer_weights + args.kernel_rows * n_dots_per_kernel_row * 4 * iter_length); in pack_parameters()
115 for (unsigned int ki = 0; ki < args.kernel_rows; ki++) in pack_parameters()
145 args.kernel_rows * args.kernel_cols * qp.a_offset * qp.b_offset; in pack_parameters()
|
| D | generic.hpp | 38 const unsigned int kernel_rows; member
48 unsigned int kernel_points(void) const { return kernel_cols * kernel_rows; } in kernel_points()
51 unsigned int kernel_rows,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_conv2d.cc | 25 int64_t input_cols, int64_t input_channels, int64_t kernel_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16() argument
34 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
45 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32() argument
54 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
|
| D | runtime_conv2d.cc | 28 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenConv2DF32() argument
40 input_rows, input_cols, input_channels, kernel_rows, 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() argument
62 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConv2DF16()
|
| D | runtime_conv2d_acl.cc | 32 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in ACLDepthwiseConvImpl() argument
68 arm_compute::TensorShape(kernel_filters, kernel_cols, kernel_rows), 1, in ACLDepthwiseConvImpl()
130 int64_t kernel_rows, int64_t kernel_cols, in ACLGemmConvImpl() argument
169 arm_compute::TensorShape(input_channels, kernel_cols, kernel_rows, in ACLGemmConvImpl()
236 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_ACLConv2DF32() argument
252 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_ACLConv2DF32()
260 input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_ACLConv2DF32()
273 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_ACLConv2DF32()
|
| D | runtime_conv2d_mkl.cc | 55 int64_t kernel_rows, int64_t kernel_cols, in MKLConvImpl() argument
73 ToInt(kernel_channels), ToInt(kernel_rows), in MKLConvImpl()
157 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_MKLConv2DF32() argument
169 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConv2DF32()
175 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConv2DF32()
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/ |
| D | sve_fp32_packed_to_nhwc_3x3_s2_with_multiplier_output3x3_mla_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 3; member
48 : Parent(3, 3, kernel_rows, kernel_cols, stride_rows, stride_cols) in sve_fp32_packed_to_nhwc_3x3_s2_with_multiplier_output3x3_mla_depthfirst()
|
| D | a64_fp32_packed_to_nhwc_3x3_s2_with_multiplier_output3x3_mla_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 3; member
48 : Parent(3, 3, kernel_rows, kernel_cols, stride_rows, stride_cols) in a64_fp32_packed_to_nhwc_3x3_s2_with_multiplier_output3x3_mla_depthfirst()
|
| D | a64_fp32_packed_to_nhwc_5x5_s1_with_multiplier_output2x4_mla_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 5; member
48 : Parent(2, 4, kernel_rows, kernel_cols, stride_rows, stride_cols) in a64_fp32_packed_to_nhwc_5x5_s1_with_multiplier_output2x4_mla_depthfirst()
|
| D | sve_fp32_packed_to_nhwc_5x5_s1_with_multiplier_output2x4_mla_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 5; member
48 : Parent(2, 4, kernel_rows, kernel_cols, stride_rows, stride_cols) in sve_fp32_packed_to_nhwc_5x5_s1_with_multiplier_output2x4_mla_depthfirst()
|
| D | a64_u8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 3; member
48 : Parent(2, 4, kernel_rows, kernel_cols, stride_rows, stride_cols) in a64_u8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst()
|
| D | sve_u8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 5; member
48 : Parent(4, 2, kernel_rows, kernel_cols, stride_rows, stride_cols) in sve_u8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst()
|
| D | a64_s8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 3; member
48 : Parent(2, 4, kernel_rows, kernel_cols, stride_rows, stride_cols) in a64_s8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst()
|
| D | sve_u8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 3; member
48 : Parent(2, 4, kernel_rows, kernel_cols, stride_rows, stride_cols) in sve_u8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst()
|
| D | a64_s8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 5; member
48 : Parent(4, 2, kernel_rows, kernel_cols, stride_rows, stride_cols) in a64_s8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst()
|
| D | a64_u8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 5; member
48 : Parent(4, 2, kernel_rows, kernel_cols, stride_rows, stride_cols) in a64_u8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst()
|
| D | sve_s8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 3; member
48 : Parent(2, 4, kernel_rows, kernel_cols, stride_rows, stride_cols) in sve_s8q_packed_to_nhwc_3x3_s2_with_multiplier_output2x4_dot_depthfirst()
|
| D | sve_s8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst.hpp | 41 constexpr static unsigned int kernel_rows = 5; member
48 : Parent(4, 2, kernel_rows, kernel_cols, stride_rows, stride_cols) in sve_s8q_packed_to_nhwc_5x5_s1_with_multiplier_output4x2_dot_depthfirst()
|
| 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
61 : Parent(kernel_rows, kernel_cols, stride_rows, stride_cols, output_rows, vl_type) in sme2_fp32_planar_3x3_s2_4rows_mla_za()
|
| 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
61 : Parent(kernel_rows, kernel_cols, stride_rows, stride_cols, output_rows, vl_type) in sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za()
|
| /external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| D | depthwise.hpp | 49 unsigned int kernel_rows, kernel_cols; member
66 unsigned int kernel_rows, unsigned int kernel_cols, in DepthwiseArgs()
74 …: cpu_info(cpu_info), kernel_rows(kernel_rows), kernel_cols(kernel_cols), stride_rows(stride_rows)… in DepthwiseArgs()
|