depthwise_conv_op.h File Reference

#include <thread>
#include "unsupported/Eigen/CXX11/Tensor"
#include "cker/operation/Helper/Tensor.h"

Go to the source code of this file.

Data Structures
struct	nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvOp< Device, T >
struct	nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvBackpropInputOp< Device, T >
struct	nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvBackpropFilterOp< Device, T >
struct	nnfw::cker::depthwise_conv_op::functor::DepthwiseFilterPadOp< T >
struct	nnfw::cker::depthwise_conv_op::functor::DepthwiseInputCopyOp< T >
struct	nnfw::cker::depthwise_conv_op::DepthwiseConv2DKernel< T >
struct	nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvOp< CPUDevice, T >
struct	nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvBackpropInputOp< CPUDevice, T >
struct	nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvBackpropFilterOp< CPUDevice, T >

Namespaces
namespace	nnfw
namespace	nnfw::cker
namespace	nnfw::cker::depthwise_conv_op
namespace	nnfw::cker::depthwise_conv_op::functor

Macros
#define	EIGEN_USE_THREADS

Typedefs
using	nnfw::cker::depthwise_conv_op::CPUDevice = Eigen::ThreadPoolDevice

Functions
template<typename T >
void	nnfw::cker::depthwise_conv_op::CopyOutputBackpropRegion (int, int, int, int, int filter_rows_, int filter_cols_, int, int stride_, int pad_rows_, int pad_cols_, int out_rows_, int out_cols_, int out_depth, const int64_t padded_filter_inner_dim_size, const int64_t in_r, const int64_t in_c, const T *out_backprop, T *buffer)
template<typename T >
void	nnfw::cker::depthwise_conv_op::ComputeBackpropInput (int, int, int in_cols, int in_depth_, int filter_rows, int filter_cols, int depth_multiplier_, int, int, int, int, int, int out_depth_, const int64_t padded_filter_inner_dim_size, const int64_t in_r, const int64_t in_c, const T *filter, const T *buffer, T *out_buffer, T *output)
template<typename T >
void	nnfw::cker::depthwise_conv_op::DepthwiseConvBackpropInputReference (int batch, int in_rows, int in_cols, int in_depth, int out_rows, int out_cols, int out_depth, int stride, int depth_multiplier, int filter_rows, int filter_cols, int pad_rows, int pad_cols, const T *out_backprop, const T *filter, T *in_backprop)
template<typename T >
void	nnfw::cker::depthwise_conv_op::ComputeBackpropFilter (int, int, int, int, int filter_rows, int filter_cols, int, int, int, int, int out_rows, int out_cols, int out_depth_, const int64_t padded_out_depth_size, const int64_t out_r, const int64_t out_c, const T *out_backprop, const T *input_buffer, T *output_buffer)
template<typename T >
void	nnfw::cker::depthwise_conv_op::DepthwiseConvBackpropFilterReference (int batch, int in_rows, int in_cols, int in_depth, int out_rows, int out_cols, int out_depth, int stride, int depth_multiplier, int filter_rows, int filter_cols, int pad_rows, int pad_cols, const T *out_backprop, const T *input, T *filter_backprop)

Macro Definition Documentation

EIGEN_USE_THREADS

#define EIGEN_USE_THREADS

Definition at line 22 of file depthwise_conv_op.h.