DLPrimitives
Static Protected Member Functions | List of all members
dlprim::Convolution2DBase Class Reference
Inheritance diagram for dlprim::Convolution2DBase:
dlprim::Convolution2D dlprim::TransposedConvolution2D

Static Protected Member Functions

template<typename Op , typename DType >
static void im2col (Shape const &in, Shape const &outs, DType *img_in, DType *mat_in, Convolution2DConfig const &config)
 
static void fwd_bwd_cpu (GemmOpMode mode, Tensor &in, Tensor &out, Tensor &W, Tensor *bias_tensor, void *ws, Convolution2DConfig const &config, float fwd_beta=0.0f)
 
static void scale_cpu (Tensor &t, float v)
 

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