DLPrimitives
|
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) |