24.02.1
|
Go to the documentation of this file.
62 template <
typename TInput,
typename TWeight,
typename TOutput,
typename TAccum,
typename OutputStage>
72 [
this] (
unsigned int idx,
unsigned int &x,
unsigned int &y) ->
bool
79 const DepthwiseArgs &
args,
void *buffer,
80 const void *biases,
const OutputStage &,
81 const void *weights,
size_t ld_weight_col,
size_t ld_weight_row
88 [
this] (
unsigned int idx,
unsigned int &x,
unsigned int &y) ->
bool
92 packing_args,
args, buffer, biases, weights, ld_weight_col, ld_weight_row);
virtual unsigned int get_n_input_points() const
template UniqueDepthwiseCommon< float > depthwise(const DepthwiseArgs &, const Nothing &)
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)
virtual unsigned int get_n_kernel_points() const
size_t get_storage_size_generic(const PackingArguments &packing_args, const DepthwiseArgs &args)
virtual unsigned int get_stride_cols() const =0
virtual unsigned int get_kernel_rows() const =0
virtual unsigned int get_n_output_points() const
virtual unsigned int get_input_cols() const override
virtual void pack_parameters(const DepthwiseArgs &args, void *buffer, const void *biases, const OutputStage &, const void *weights, size_t ld_weight_col, size_t ld_weight_row) const
virtual size_t get_storage_size(const DepthwiseArgs &args) const
virtual unsigned int get_input_rows() const override
virtual bool uses_premultiply() const
virtual arm_gemm::VLType get_vl_type() const =0
virtual bool get_kernel_packing_point(const unsigned int index, unsigned int &x, unsigned int &y) const
virtual unsigned int get_stride_rows() const =0
virtual unsigned int get_kernel_cols() const =0
virtual unsigned int get_accumulator_depth_vl() const