Skip to content
Snippets Groups Projects

feat_operator_convtranspose

Merged Grégoire Kubler requested to merge feat_operator_convtranspose into dev
3 files
+ 213
86
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -200,6 +200,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.
*
Loading