ONE - On-device Neural Engine
Loading...
Searching...
No Matches
DepthwiseConv.h File Reference
#include "cker/eigen/depthwise_conv_op.h"
#include "cker/Shape.h"
#include "cker/Types.h"

Go to the source code of this file.

Namespaces

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

Functions

template<typename T >
void nnfw::cker::train::backpropInput (const DepthwiseConvParams &params, const Shape &incoming_shape, const T *incoming_data, const Shape &filter_shape, const T *filter_data, T *padded_filter_data, const Shape &grad_shape, T *grad_data, bool pad_filter, T *filter_buffers_data, T *filter_dim_buffers_data)
 
template<typename T >
void nnfw::cker::train::backpropFilter (const DepthwiseConvParams &params, const Shape &incoming_shape, const T *incoming_data, const Shape &input_shape, const T *input_data, const Shape &filter_grad_shape, T *filter_grad_data, T *padded_filter_data, T *filter_buffers_data)