24.02.1
|
#include <depthwise_strategies_common.hpp>
Public Member Functions | |
virtual arm_gemm::VLType | get_vl_type () const =0 |
virtual unsigned int | get_kernel_rows () const =0 |
virtual unsigned int | get_kernel_cols () const =0 |
virtual unsigned int | get_stride_rows () const =0 |
virtual unsigned int | get_stride_cols () const =0 |
virtual unsigned int | get_input_rows () const override |
virtual unsigned int | get_input_cols () const override |
virtual unsigned int | get_n_input_points () const |
virtual unsigned int | get_n_output_points () const |
virtual unsigned int | get_n_kernel_points () const |
virtual bool | uses_premultiply () const |
virtual unsigned int | get_accumulator_depth_vl () const |
virtual bool | get_kernel_packing_point (const unsigned int index, unsigned int &x, unsigned int &y) const |
Public Member Functions inherited from IDepthfirstStrategy | |
virtual | ~IDepthfirstStrategy ()=default |
virtual unsigned int | get_output_rows () const =0 |
virtual unsigned int | get_output_cols () const =0 |
Definition at line 34 of file depthwise_strategies_common.hpp.
|
virtual |
Reimplemented in GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >.
Definition at line 46 of file depthwise_strategies_common.cpp.
Referenced by DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), and DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters().
|
overridevirtual |
Implements IDepthfirstStrategy.
Definition at line 35 of file depthwise_strategies_common.cpp.
References DepthfirstStrategyUntyped::get_kernel_cols(), IDepthfirstStrategy::get_output_cols(), and DepthfirstStrategyUntyped::get_stride_cols().
Referenced by DepthfirstStrategyUntyped::get_n_input_points().
|
overridevirtual |
Implements IDepthfirstStrategy.
Definition at line 30 of file depthwise_strategies_common.cpp.
References DepthfirstStrategyUntyped::get_kernel_rows(), IDepthfirstStrategy::get_output_rows(), and DepthfirstStrategyUntyped::get_stride_rows().
Referenced by DepthfirstStrategyUntyped::get_n_input_points().
|
pure virtual |
Implemented in DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, TAccum, OutputStage >.
Referenced by DepthfirstStrategyUntyped::get_input_cols(), DepthfirstStrategyUntyped::get_kernel_packing_point(), DepthfirstStrategyUntyped::get_n_kernel_points(), DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), and DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters().
|
virtual |
Definition at line 48 of file depthwise_strategies_common.cpp.
References DepthfirstStrategyUntyped::get_kernel_cols(), and DepthfirstStrategyUntyped::get_kernel_rows().
Referenced by DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::get_storage_size(), DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters(), and GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::pack_parameters().
|
pure virtual |
Implemented in DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, TAccum, OutputStage >.
Referenced by DepthfirstStrategyUntyped::get_input_rows(), DepthfirstStrategyUntyped::get_kernel_packing_point(), DepthfirstStrategyUntyped::get_n_kernel_points(), DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), and DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters().
|
virtual |
Definition at line 40 of file depthwise_strategies_common.cpp.
References DepthfirstStrategyUntyped::get_input_cols(), and DepthfirstStrategyUntyped::get_input_rows().
|
virtual |
Definition at line 42 of file depthwise_strategies_common.cpp.
References DepthfirstStrategyUntyped::get_kernel_cols(), and DepthfirstStrategyUntyped::get_kernel_rows().
|
virtual |
Definition at line 41 of file depthwise_strategies_common.cpp.
References IDepthfirstStrategy::get_output_cols(), and IDepthfirstStrategy::get_output_rows().
|
pure virtual |
Implemented in DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, TAccum, OutputStage >.
Referenced by DepthfirstStrategyUntyped::get_input_cols().
|
pure virtual |
Implemented in DepthwiseDepthfirstStrategyCommon< TInput, TWeight, TOutput, TAccum, OutputStage >.
Referenced by DepthfirstStrategyUntyped::get_input_rows().
|
pure virtual |
Implemented in GenericDepthfirstMultiplierStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >, GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >, sme2_fp16_nhwc_3x3_s1_output2x2_mla_depthfirst, sme2_fp16_nhwc_3x3_s1_output3x3_mla_depthfirst, sme2_fp16_nhwc_3x3_s1_output4x4_mla_depthfirst, sme2_fp16_nhwc_3x3_s2_output2x2_mla_depthfirst, and sme2_fp16_nhwc_5x5_s1_output2x2_mla_depthfirst.
Referenced by DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), and DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters().
|
virtual |
Definition at line 44 of file depthwise_strategies_common.cpp.
Referenced by DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::get_storage_size(), GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::get_storage_size(), DepthfirstStrategy< __fp16, __fp16, __fp16, __fp16, typename DefaultOutputStage< __fp16 >::Type >::pack_parameters(), and GenericDepthfirstStrategy< TInput, TWeight, TOutput, TAccum, OutputStage >::pack_parameters().