Skip to content
Snippets Groups Projects

feat_operator_convtranspose

Merged Grégoire Kubler requested to merge feat_operator_convtranspose into dev
2 files
+ 25
3
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -12,6 +12,7 @@
#ifndef AIDGE_CORE_OPERATOR_CONV_H_
#define AIDGE_CORE_OPERATOR_CONV_H_
#include <algorithm>
#include <array>
#include <cstddef> // std::size_t
#include <string>
@@ -234,10 +235,16 @@ std::shared_ptr<Node> Conv(DimSize_t inChannels,
bool noBias = false);
/**
* @brief Helper function for Conv with C-style arrays.
* @brief Perform a convolution on the input Tensor.
*
* This helper function allows automatic template deduction of the number of dimensions (DIM)
* based on the kernel dimensions provided.
* @tparam DIM Number of dimensions for the feature map.
* @param inChannels Number of input channels.
* @param outChannels Number of output channels.
* @param kernelDims Dimensions of the kernel. Must be the same number of dimensions as the feature map.
* @param name Name of the operator.
* @param strideDims Dimensions of the stride attribute. Must be the same number of dimensions as the feature map.
* @param dilationDims Dimensions of the dilation attribute. Must be the same number of dimensions as the feature map.
* @return std::shared_ptr<Node> A Node containing the operator.
*/
template <DimSize_t DIM>
std::shared_ptr<Node> Conv(
Loading