Searched refs:packing_args (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/ |
| H A D | generic.cpp | 45 size_t get_storage_size_generic(const PackingArguments &packing_args, const DepthwiseArgs &args) in get_storage_size_generic() argument
55 return args.input_channels * get_storage_size_generic(packing_args, args_per_input_channel); in get_storage_size_generic()
59 packing_args.accumulator_depth_vl * in get_storage_size_generic()
60 …arm_gemm::utils::get_vector_length<uint8_t>(packing_args.vl_type) / packing_args.accumulator_eleme… in get_storage_size_generic()
62 const auto pack_size = (packing_args.include_bias ? packing_args.bias_element_size : 0) + in get_storage_size_generic()
63 packing_args.kernel_points() * packing_args.weight_element_size; in get_storage_size_generic()
68 const PackingArguments &packing_args, in pack_parameters_generic() argument
93 ld_weight_row = ld_weight_row ? ld_weight_row : ld_weight_col * packing_args.kernel_cols; in pack_parameters_generic()
95 auto per_input_channel_size = get_storage_size_generic(packing_args, args_per_input_channel); in pack_parameters_generic()
100 … packing_args, args_per_input_channel, buffer, biases, weights, ld_weight_col, ld_weight_row); in pack_parameters_generic()
[all …]
|
| H A D | generic.hpp | 64 const PackingArguments &packing_args,
69 const PackingArguments &packing_args,
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| H A D | depthwise_strategies_common.hpp | 66 interleaves::PackingArguments packing_args( in get_storage_size() local
73 return interleaves::get_storage_size_generic(packing_args, args); in get_storage_size()
82 interleaves::PackingArguments packing_args( in pack_parameters() local
90 packing_args, args, buffer, biases, weights, ld_weight_col, ld_weight_row); in pack_parameters()
|
| H A D | depthwise_depthfirst_generic.hpp | 100 interleaves::PackingArguments packing_args( in get_storage_size() local
107 return interleaves::get_storage_size_generic(packing_args, args); in get_storage_size()
116 interleaves::PackingArguments packing_args( in pack_parameters() local
124 packing_args, args, buffer, biases, weights, ld_weight_col, ld_weight_row); in pack_parameters()
|