Skip to content
Snippets Groups Projects

Feat/operator squeeze unsqueeze

All threads resolved!
73 files
+ 852
300
Compare changes
  • Side-by-side
  • Inline
Files
73
@@ -12,6 +12,7 @@ import aidge_core
@@ -12,6 +12,7 @@ import aidge_core
import onnx
import onnx
from onnx import helper
from onnx import helper
from aidge_onnx.node_export import auto_register_export
from aidge_onnx.node_export import auto_register_export
 
from typing import List
@auto_register_export("Add")
@auto_register_export("Add")
def export_add(
def export_add(
@@ -20,7 +21,7 @@ def export_add(
@@ -20,7 +21,7 @@ def export_add(
node_outputs_name,
node_outputs_name,
opset:int = None,
opset:int = None,
verbose: bool = False,
verbose: bool = False,
**kwargs) -> None:
**kwargs) -> List[helper.NodeProto]:
aidge_operator = aidge_node.get_operator()
aidge_operator = aidge_node.get_operator()
Loading