nnfw::cker::train::LaunchConv2DBackpropInputOp< T > Struct Template Reference

#include <Conv.h>

Public Member Functions
void	operator() (const T *out_backprop_data, int batches, int out_backprop_height, int out_backprop_width, int output_depth, const T *filter_data, int filter_height, int filter_width, int row_dilation, int col_dilation, int row_stride, int col_stride, const PaddingType &padding_type, int padding_top, int padding_bottom, int padding_left, int padding_right, T *in_backprop_data, int in_backprop_height, int in_backprop_width, int input_depth)

Detailed Description

template<typename T>
struct nnfw::cker::train::LaunchConv2DBackpropInputOp< T >

Definition at line 122 of file Conv.h.

Member Function Documentation

operator()()

template<typename T >

void nnfw::cker::train::LaunchConv2DBackpropInputOp< T >::operator() ( const T *  out_backprop_data, int  batches, int  out_backprop_height, int  out_backprop_width, int  output_depth, const T *  filter_data, int  filter_height, int  filter_width, int  row_dilation, int  col_dilation, int  row_stride, int  col_stride, const PaddingType &  padding_type, int  padding_top, int  padding_bottom, int  padding_left, int  padding_right, T *  in_backprop_data, int  in_backprop_height, int  in_backprop_width, int  input_depth  )

inline

Definition at line 124 of file Conv.h.

  {
    LaunchConv2DBackpropInputOpImpl<Eigen::ThreadPoolDevice, T> launcher;
    launcher(*eigen_support::GetThreadPoolDevice(), out_backprop_data, batches, out_backprop_height,
             out_backprop_width, output_depth, filter_data, filter_height, filter_width,
             row_dilation, col_dilation, row_stride, col_stride, padding_type, padding_top,
             padding_bottom, padding_left, padding_right, in_backprop_data, in_backprop_height,
             in_backprop_width, input_depth);
  }

References nnfw::cker::eigen_support::GetThreadPoolDevice().

---

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

• compute/cker/include/cker/train/operation/Conv.h