Skip to content
Snippets Groups Projects

feat_operator_convtranspose

Merged Grégoire Kubler requested to merge feat_operator_convtranspose into dev
Files
11
@@ -13,45 +13,64 @@
@@ -13,45 +13,64 @@
#define AIDGE_CPU_OPERATOR_CONVIMPL_H_
#define AIDGE_CPU_OPERATOR_CONVIMPL_H_
#include <array>
#include <array>
#include <memory>
#include <tuple>
#include <vector>
#include "aidge/backend/cpu/operator/OperatorImpl.hpp"
#include "aidge/backend/cpu/operator/OperatorImpl.hpp"
#include "aidge/operator/Conv.hpp"
#include "aidge/operator/Conv.hpp"
#include "aidge/utils/Registrar.hpp"
#include "aidge/utils/Registrar.hpp"
#include "aidge/utils/Types.h"
#include "aidge/utils/Types.h"
#include "aidge/backend/cpu/data/GetCPUPtr.h"
namespace Aidge {
namespace Aidge {
 
// Operator implementation entry point for the backend
// Operator implementation entry point for the backend
using Conv1D_Op = Conv_Op<1>;
using Conv1D_Op = Conv_Op<1>;
using ConvImpl1D_cpu = OperatorImpl_cpu<Conv_Op<1>,
using ConvImpl1D_cpu = OperatorImpl_cpu<Conv_Op<1>,
void(const std::array<DimSize_t, 1>&,
void(const std::array<DimSize_t, 1> &,
const std::array<DimSize_t, 1>&,
const std::array<DimSize_t, 1> &,
const std::array<DimSize_t, 1>&,
const std::array<DimSize_t, 1> &,
const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 3> &,
DimSize_t,
DimSize_t,
const void *,
const void *,
const void *,
const void *,
const void *,
const void *,
void *)>;
void *),
 
void(const std::array<DimSize_t, 1> &,
 
const std::array<DimSize_t, 1> &,
 
const std::array<DimSize_t, 1> &,
 
const std::array<DimSize_t, 3> &,
 
const std::array<DimSize_t, 3> &,
 
const void *,
 
const void *,
 
const void *,
 
void *,
 
void *,
 
void *)>;
using Conv2D_Op = Conv_Op<2>;
using Conv2D_Op = Conv_Op<2>;
using ConvImpl2D_cpu = OperatorImpl_cpu<Conv_Op<2>,
using ConvImpl2D_cpu = OperatorImpl_cpu<Conv2D_Op,
void(const std::array<DimSize_t, 2>&,
void(const std::array<DimSize_t, 2> &,
const std::array<DimSize_t, 2>&,
const std::array<DimSize_t, 2> &,
const std::array<DimSize_t, 2>&,
const std::array<DimSize_t, 2> &,
const std::array<DimSize_t, 4> &,
const std::array<DimSize_t, 4> &,
DimSize_t,
DimSize_t,
const void *,
const void *,
const void *,
const void *,
const void *,
const void *,
void *)>;
void *),
 
void(const std::array<DimSize_t, 2> &,
 
const std::array<DimSize_t, 2> &,
 
const std::array<DimSize_t, 2> &,
 
const std::array<DimSize_t, 4> &,
 
const std::array<DimSize_t, 4> &,
 
const void *,
 
const void *,
 
const void *,
 
void *,
 
void *,
 
void *)>;
// Implementation entry point registration to Operator
// Implementation entry point registration to Operator
REGISTRAR(Conv1D_Op, "cpu", Aidge::ConvImpl1D_cpu::create);
REGISTRAR(Conv1D_Op, "cpu", Aidge::ConvImpl1D_cpu::create);
REGISTRAR(Conv2D_Op, "cpu", Aidge::ConvImpl2D_cpu::create);
REGISTRAR(Conv2D_Op, "cpu", Aidge::ConvImpl2D_cpu::create);
} // namespace Aidge
} // namespace Aidge
#endif /* AIDGE_CPU_OPERATOR_CONVIMPL_H_ */
#endif /* AIDGE_CPU_OPERATOR_CONVIMPL_H_ */
Loading