Skip to content
Snippets Groups Projects

Multiple refactors

Merged Olivier BICHLER requested to merge nobias into dev
Files
15
@@ -25,6 +25,29 @@
@@ -25,6 +25,29 @@
namespace Aidge {
namespace Aidge {
// class ConvDepthWise_Op;
// 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
// compute kernel registry for forward and backward
class ConvDepthWiseImpl2DForward_cpu
class ConvDepthWiseImpl2DForward_cpu
Loading