24.02.1
|
Namespaces | |
quantized | |
Data Structures | |
struct | PackingArguments |
Functions | |
size_t | get_storage_size_generic (const PackingArguments &packing_args, const DepthwiseArgs &args) |
void | pack_parameters_generic (const PackingArguments &packing_args, const DepthwiseArgs &args, void *buffer_raw, const void *biases_raw, const void *weights_raw, size_t ld_weight_col, size_t ld_weight_row) |
size_t get_storage_size_generic | ( | const PackingArguments & | packing_args, |
const DepthwiseArgs & | args | ||
) |
Definition at line 45 of file generic.cpp.
References PackingArguments::accumulator_depth_vl, PackingArguments::accumulator_element_size, GemmTuner::args, PackingArguments::bias_element_size, arm_gemm::iceildiv(), PackingArguments::include_bias, PackingArguments::kernel_points(), PackingArguments::premultiply, PackingArguments::vl_type, and PackingArguments::weight_element_size.
Referenced by DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), DepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum >::get_storage_size(), GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::get_storage_size(), DepthwiseDepthfirstStrategy< TInput, TWeight, TOutput, int32_t >::get_storage_size(), PlanarStrategy< uint8_t, int8_t >::get_storage_size(), GenericDepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::get_storage_size(), and pack_parameters_generic().
void pack_parameters_generic | ( | const PackingArguments & | packing_args, |
const DepthwiseArgs & | args, | ||
void * | buffer_raw, | ||
const void * | biases_raw, | ||
const void * | weights_raw, | ||
size_t | ld_weight_col, | ||
size_t | ld_weight_row | ||
) |
Definition at line 67 of file generic.cpp.
References PackingArguments::accumulator_depth_vl, PackingArguments::accumulator_element_size, GemmTuner::args, PackingArguments::bias_element_size, get_storage_size_generic(), PackingArguments::get_weight_pos, PackingArguments::include_bias, PackingArguments::kernel_cols, PackingArguments::premultiply, PackingArguments::vl_type, and PackingArguments::weight_element_size.
Referenced by DepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum >::pack_parameters(), DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters(), GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::pack_parameters(), DepthwiseDepthfirstStrategy< TInput, TWeight, TOutput, int32_t >::pack_parameters(), PlanarStrategy< uint8_t, int8_t >::pack_parameters(), and GenericDepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::pack_parameters().