Skip to content
Snippets Groups Projects
Commit 7fef0535 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Reflect changes due to resaling kernel moved.

parent e14e2c79
No related branches found
No related tags found
3 merge requests!27v0.2.0,!22v0.4.0,!15Export refactor
......@@ -5,7 +5,6 @@ class ExportLibCpp(ExportLib):
name="export_cpp"
static_files={
str(ROOT / "static" / "Makefile"): "",
str(ROOT / "static" / "include" / "network" / "rescaling.hpp"): "dnn/include/network",
str(ROOT / "static" / "include" / "network" / "typedefs.hpp"): "dnn/include/network",
str(ROOT / "static" / "include" / "network" / "utils.hpp"): "dnn/include/network",
}
......@@ -94,6 +94,7 @@ class ReLUCPP(ExportNodeCpp):
self.include_list = []
self.kernels_to_copy = [
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
......@@ -118,6 +119,7 @@ class ConvCPP(ExportNodeCpp):
str(ROOT / "kernels" / "convolution.hpp"),
str(ROOT / "kernels" / "macs.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
......@@ -152,6 +154,7 @@ class PaddedConvCPP(ExportNodeCpp):
str(ROOT / "kernels" / "convolution.hpp"),
str(ROOT / "kernels" / "macs.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
......@@ -172,6 +175,7 @@ class AddCPP(ExportNodeCpp):
self.kernels_to_copy = [
str(ROOT / "kernels" / "elemwise.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
......@@ -192,6 +196,7 @@ class SubCPP(ExportNodeCpp):
self.kernels_to_copy = [
str(ROOT / "kernels" / "elemwise.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
......@@ -213,6 +218,7 @@ class MulCPP(ExportNodeCpp):
self.kernels_to_copy = [
str(ROOT / "kernels" / "elemwise.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
......@@ -316,6 +322,7 @@ class FcCPP(ExportNodeCpp):
str(ROOT / "kernels" / "fullyconnected.hpp"),
str(ROOT / "kernels" / "macs.hpp"),
str(ROOT / "kernels" / "activation.hpp"),
str(ROOT / "kernels" / "rescaling.hpp")
]
@classmethod
def exportable(cls, node):
......
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