DLPrimitives
|
Perform InnerProduct/FullyConnected/Dense backward data calculations. More...
#include <include/dlprim/core/conv.hpp>
Public Member Functions | |
virtual void | enqueue (Tensor &dx, Tensor &w, Tensor &dy, Tensor &ws, float factor, ExecutionContext const &e)=0 |
Public Member Functions inherited from dlprim::core::Conv2DBase | |
virtual char const * | algo () const =0 |
virtual size_t | workspace () |
Static Public Member Functions | |
static std::unique_ptr< Conv2DBackwardData > | create (Context &ctx, Conv2DSettings const &config, std::string const &algo=std::string()) |
Static Public Member Functions inherited from dlprim::core::Conv2DBase | |
static Shape | get_output_shape (Convolution2DConfigBase const &config, Shape const &in) |
static Shape | get_output_shape_transposed (Convolution2DConfigBase const &config, Shape const &in, int output_pad[2]) |
Perform InnerProduct/FullyConnected/Dense backward data calculations.