ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Conv.h File Reference

Go to the source code of this file.

Data Structures

struct  nnfw::cker::train::functor::SpatialConvolutionBackwardInputFunc< Device, T >
 
struct  nnfw::cker::train::functor::SpatialConvolutionBackwardInputWithExplicitPaddingFunc< Device, T >
 
struct  nnfw::cker::train::LaunchConv2DBackpropInputOpImpl< Device, T >
 
struct  nnfw::cker::train::LaunchConv2DBackpropInputOp< T >
 
struct  nnfw::cker::train::LaunchConv2DBackpropFilterOp< T >
 

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 
namespace  nnfw::cker::train
 
namespace  nnfw::cker::train::functor
 

Functions

void nnfw::cker::train::ConvInputGrad (const ConvParams &params, const Shape &incoming_shape, const float *incoming_data, const Shape &filter_shape, const float *filter_data, const int padding_bottom, const int padding_right, const Shape &grad_shape, float *grad_data)
 
void nnfw::cker::train::ConvFilterGrad (const ConvParams &params, const Shape &incoming_shape, const float *incoming_data, const Shape &input_shape, const float *input_data, const int padding_bottom, const int padding_right, const Shape &filter_backprop_shape, float *filter_backprop_data)