Skip to content
Snippets Groups Projects

Export refactor

Merged Cyril Moineau requested to merge allowNoInputProducer into dev
Compare and
100 files
+ 1623
567
Compare changes
  • Side-by-side
  • Inline
Files
100
 
from aidge_core.aidge_export_aidge.registry import ExportSerialize
 
from aidge_core.aidge_export_aidge import ROOT_EXPORT
 
from aidge_core.export_utils import ExportNodeCpp
 
from aidge_core import ImplSpec, IOSpec, dtype
 
 
@ExportSerialize.register("Add", ImplSpec(IOSpec(dtype.any)))
 
class Add(ExportNodeCpp):
 
def __init__(self, node, mem_info):
 
super().__init__(node, mem_info)
 
self.config_template = ""
 
self.forward_template = str(
 
ROOT_EXPORT / "templates/graph_ctor/add.jinja")
 
self.include_list = ["aidge/operator/Add.hpp"]
 
self.kernels_to_copy = []
Loading