Skip to content
Snippets Groups Projects

feat_operator_convtranspose

Merged Grégoire Kubler requested to merge feat_operator_convtranspose into dev
3 files
+ 286
2
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -211,6 +211,48 @@ PaddedAvgPooling(DimSize_t const (&kernel_dims)[DIM],
@@ -211,6 +211,48 @@ PaddedAvgPooling(DimSize_t const (&kernel_dims)[DIM],
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
 
template <std::array<DimSize_t, 1>::size_type DIM>
 
extern std::shared_ptr<Node> PaddedConvTranspose(
 
const DimSize_t &inChannels,
 
const DimSize_t &outChannels,
 
const std::array<DimSize_t, DIM> &kernelDims,
 
const std::array<DimSize_t, DIM> &strideDims =
 
create_array<DimSize_t,DIM>(1),
 
const std::array<DimSize_t, DIM> &dilationDims =
 
create_array<DimSize_t,DIM>(1),
 
const bool noBias = false,
 
const std::array<DimSize_t, 2*DIM> &paddingDims =
 
create_array<DimSize_t,2*DIM>(0),
 
const std::string& name = "");
 
 
template <std::array<DimSize_t, 1>::size_type DIM>
 
extern std::shared_ptr<Node> PaddedConvTranspose(const DimSize_t &inChannels,
 
const DimSize_t &outChannels,
 
DimSize_t const (&kernel_dims)[DIM],
 
const std::array<DimSize_t, DIM> &strideDims =
 
create_array<DimSize_t,DIM>(1),
 
const std::array<DimSize_t, DIM> &dilationDims =
 
create_array<DimSize_t,DIM>(1),
 
const bool noBias = false,
 
const std::array<DimSize_t, 2*DIM> &paddingDims =
 
create_array<DimSize_t,2*DIM>(0),
 
const std::string& name = "");
 
 
template <std::array<DimSize_t, 1>::size_type DIM>
 
extern std::shared_ptr<MetaOperator_Op> PaddedConvTranspose_Op(const DimSize_t &inChannels,
 
const DimSize_t &outChannels,
 
const std::array<DimSize_t, DIM> &kernelDims,
 
const std::array<DimSize_t, DIM> &strideDims =
 
create_array<DimSize_t,DIM>(1),
 
const std::array<DimSize_t, DIM> &dilationDims =
 
create_array<DimSize_t,DIM>(1),
 
const bool noBias = false,
 
const std::array<DimSize_t, 2*DIM> &paddingDims =
 
create_array<DimSize_t,2*DIM>(0),
 
const std::string& name = "");
 
 
////////////////////////////////////////////////////////////////////////////////
 
/**
/**
* @brief Creates a padded max pooling operation.
* @brief Creates a padded max pooling operation.
*
*
Loading