Skip to content
Snippets Groups Projects

feat : support for conv3D forward

Merged Grégoire Kubler requested to merge feat_operator_conv3D into dev
12 files
+ 4080
836
Compare changes
  • Side-by-side
  • Inline
Files
12
@@ -68,9 +68,34 @@ using ConvImpl2D_cpu = OperatorImpl_cpu<Conv2D_Op,
void *,
void *)>;
using Conv3D_Op = Conv_Op<3>;
using ConvImpl3D_cpu = OperatorImpl_cpu<Conv3D_Op,
void(const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 5> &,
const std::array<DimSize_t, 5> &,
const void *,
const void *,
const void *,
void *),
void(const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 3> &,
const std::array<DimSize_t, 5> &,
const std::array<DimSize_t, 5> &,
const void *,
const void *,
const void *,
void *,
void *,
void *)>;
// Implementation entry point registration to Operator
REGISTRAR(Conv1D_Op, "cpu", Aidge::ConvImpl1D_cpu::create);
REGISTRAR(Conv2D_Op, "cpu", Aidge::ConvImpl2D_cpu::create);
REGISTRAR(Conv3D_Op, "cpu", Aidge::ConvImpl3D_cpu::create);
} // namespace Aidge
#endif /* AIDGE_CPU_OPERATOR_CONVIMPL_H_ */
Loading