nnfw::cker::DepthwiseConvWorkerTask< T, TS > Struct Template Reference

#include <DepthwiseConv.h>

Collaboration diagram for nnfw::cker::DepthwiseConvWorkerTask< T, TS >:

Public Member Functions
	DepthwiseConvWorkerTask (const DepthwiseConvParams &params, const Shape &input_shape, const T *input_data, const Shape &filter_shape, const T *filter_data, const Shape &bias_shape, const TS *bias_data, const Shape &output_shape, T *output_data, int thread_start, int thread_end, int thread_dim)
void	Run () override

Detailed Description

template<typename T, typename TS>
struct nnfw::cker::DepthwiseConvWorkerTask< T, TS >

Definition at line 45 of file DepthwiseConv.h.

Constructor & Destructor Documentation

DepthwiseConvWorkerTask()

template<typename T , typename TS >

nnfw::cker::DepthwiseConvWorkerTask< T, TS >::DepthwiseConvWorkerTask ( const DepthwiseConvParams &  params, const Shape &  input_shape, const T *  input_data, const Shape &  filter_shape, const T *  filter_data, const Shape &  bias_shape, const TS *  bias_data, const Shape &  output_shape, T *  output_data, int  thread_start, int  thread_end, int  thread_dim  )

inline

Definition at line 47 of file DepthwiseConv.h.

    : params_(params), input_shape_(input_shape), input_data_(input_data),
      filter_shape_(filter_shape), filter_data_(filter_data), bias_shape_(bias_shape),
      bias_data_(bias_data), output_shape_(output_shape), output_data_(output_data),
      thread_start_(thread_start), thread_end_(thread_end), thread_dim_(thread_dim)
  {
  }

Member Function Documentation

Run()

template<typename T , typename TS >

void nnfw::cker::DepthwiseConvWorkerTask< T, TS >::Run ( )

inlineoverride

Definition at line 58 of file DepthwiseConv.h.

  {
    optimized::DepthwiseConvImpl(params_, input_shape_, input_data_, filter_shape_, filter_data_,
                                 bias_shape_, bias_data_, output_shape_, output_data_,
                                 thread_start_, thread_end_, thread_dim_);
  }

References nnfw::cker::optimized::DepthwiseConvImpl().

---

The documentation for this struct was generated from the following file:

• compute/cker/include/cker/operation/DepthwiseConv.h