DLPrimitives
|
Static Public Member Functions | |
static TransposedConvolution2DConfig | from_json (json::value const &v) |
Static Public Member Functions inherited from dlprim::Convolution2DConfig | |
static Convolution2DConfig | from_json (json::value const &v) |
Public Attributes | |
int | output_pad [2] = {0,0} |
Public Attributes inherited from dlprim::Convolution2DConfig | |
bool | bias = true |
std::string | fwd_algo ="auto" |
std::string | bwd_data_algo ="auto" |
std::string | bwd_filter_algo ="auto" |
StandardActivations | activation =StandardActivations::identity |
Public Attributes inherited from dlprim::Convolution2DConfigBase | |
int | channels_in = -1 |
int | channels_out = -1 |
int | kernel [2] = {1,1} |
int | stride [2] = {1,1} |
int | dilate [2] = {1,1} |
int | pad [2] = {0,0} |
int | groups = 1 |