Skip to content
Snippets Groups Projects

Multiple refactors

Merged Olivier BICHLER requested to merge nobias into dev
Files
15
@@ -25,6 +25,29 @@
namespace Aidge {
// class ConvDepthWise_Op;
// compute kernel registry for forward and backward
class ConvDepthWiseImpl1DForward_cpu
: public Registrable<ConvDepthWiseImpl1DForward_cpu,
std::tuple<DataType, DataType, DataType, DataType>,
void(const ConvDepthWise_Op<1>::Attrs &, const std::array<DimSize_t, 3> &, const void *,
const void *, const void *, void *)> {};
class ConvDepthWiseImpl1D_cpu : public OperatorImpl {
public:
ConvDepthWiseImpl1D_cpu(const ConvDepthWise_Op<1> &op) : OperatorImpl(op, "cpu") {}
static std::unique_ptr<ConvDepthWiseImpl1D_cpu> create(const ConvDepthWise_Op<1> &op) {
return std::make_unique<ConvDepthWiseImpl1D_cpu>(op);
}
Elts_t getNbRequiredProtected(const IOIndex_t inputIdx) const override final;
void forward() override;
};
namespace {
// add cpu backend to ConvDepthWise_Op<1> implementation registry
static Registrar<ConvDepthWise_Op<1>> registrarConvDepthWiseImpl1D_cpu("cpu", Aidge::ConvDepthWiseImpl1D_cpu::create);
} // namespace
// compute kernel registry for forward and backward
class ConvDepthWiseImpl2DForward_cpu
Loading