Skip to content
Snippets Groups Projects
Commit bb6c36fa authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added 1D Conv and Pad

parent 2f4348a3
No related branches found
No related tags found
2 merge requests!152Update Aidge export to take a graph view has an argument instead of a...,!143Multiple refactors
Pipeline #48460 failed
...@@ -73,7 +73,7 @@ template <DimIdx_t DIM> void declare_ConvOp(py::module &m) { ...@@ -73,7 +73,7 @@ template <DimIdx_t DIM> void declare_ConvOp(py::module &m) {
void init_Conv(py::module &m) { void init_Conv(py::module &m) {
// declare_ConvOp<1>(m); declare_ConvOp<1>(m);
declare_ConvOp<2>(m); declare_ConvOp<2>(m);
// declare_ConvOp<3>(m); // declare_ConvOp<3>(m);
} }
......
...@@ -65,7 +65,7 @@ template <DimIdx_t DIM> void declare_ConvDepthWiseOp(py::module &m) { ...@@ -65,7 +65,7 @@ template <DimIdx_t DIM> void declare_ConvDepthWiseOp(py::module &m) {
void init_ConvDepthWise(py::module &m) { void init_ConvDepthWise(py::module &m) {
// declare_ConvDepthWiseOp<1>(m); declare_ConvDepthWiseOp<1>(m);
declare_ConvDepthWiseOp<2>(m); declare_ConvDepthWiseOp<2>(m);
// declare_ConvDepthWiseOp<3>(m); // declare_ConvDepthWiseOp<3>(m);
......
...@@ -180,10 +180,10 @@ void declare_LSTMOp(py::module &m) { ...@@ -180,10 +180,10 @@ void declare_LSTMOp(py::module &m) {
} }
void init_MetaOperatorDefs(py::module &m) { void init_MetaOperatorDefs(py::module &m) {
// declare_PaddedConvOp<1>(m); declare_PaddedConvOp<1>(m);
declare_PaddedConvOp<2>(m); declare_PaddedConvOp<2>(m);
// declare_PaddedConvOp<3>(m); // declare_PaddedConvOp<3>(m);
// declare_PaddedConvDepthWiseOp<1>(m); declare_PaddedConvDepthWiseOp<1>(m);
declare_PaddedConvDepthWiseOp<2>(m); declare_PaddedConvDepthWiseOp<2>(m);
// declare_PaddedConvDepthWiseOp<3>(m); // declare_PaddedConvDepthWiseOp<3>(m);
// declare_PaddedAvgPoolingOp<1>(m); // declare_PaddedAvgPoolingOp<1>(m);
......
...@@ -63,6 +63,6 @@ void init_Pad(py::module &m) { ...@@ -63,6 +63,6 @@ void init_Pad(py::module &m) {
.export_values(); .export_values();
declare_PadOp<1>(m); declare_PadOp<1>(m);
declare_PadOp<2>(m); declare_PadOp<2>(m);
declare_PadOp<3>(m); //declare_PadOp<3>(m);
} }
} // namespace Aidge } // namespace Aidge
...@@ -164,4 +164,5 @@ void Aidge::Conv_Op<DIM>::setBackend(const std::string &name, Aidge::DeviceIdx_t ...@@ -164,4 +164,5 @@ void Aidge::Conv_Op<DIM>::setBackend(const std::string &name, Aidge::DeviceIdx_t
} }
} }
template class Aidge::Conv_Op<1>;
template class Aidge::Conv_Op<2>; template class Aidge::Conv_Op<2>;
\ No newline at end of file
...@@ -164,4 +164,5 @@ void Aidge::ConvDepthWise_Op<DIM>::setBackend(const std::string &name, Aidge::De ...@@ -164,4 +164,5 @@ void Aidge::ConvDepthWise_Op<DIM>::setBackend(const std::string &name, Aidge::De
} }
} }
template class Aidge::ConvDepthWise_Op<1>;
template class Aidge::ConvDepthWise_Op<2>; template class Aidge::ConvDepthWise_Op<2>;
\ No newline at end of file
/********************************************************************************
* Copyright (c) 2023 CEA-List
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* SPDX-License-Identifier: EPL-2.0
*
********************************************************************************/
#include "aidge/operator/Pad.hpp"
template class Aidge::Pad_Op<1>;
template class Aidge::Pad_Op<2>;
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment