Skip to content
Snippets Groups Projects

feat/operator_squeeze_operator_unsqueeze

Merged Grégoire Kubler requested to merge feat/operator_squeeze_operator_unsqueeze into dev
Files
5
"""
Copyright (c) 2024 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
"""
import aidge_core
import onnx
from onnx import helper
from aidge_onnx.node_export import auto_register_export
from typing import List
@auto_register_export("Squeeze")
def export_squeeze(
aidge_node: aidge_core.Node,
node_inputs_name,
node_outputs_name,
opset : int = None,
verbose: bool = False) -> List[helper.NodeProto]:
aidge_operator = aidge_node.get_operator()
onnx_node = helper.make_node(
name=aidge_node.name(),
op_type="squeeze",
inputs=node_inputs_name,
outputs=node_outputs_name,
)
if opset is not None and opset < 13 :
onnx_node.attribute.append(
helper.make_attribute(
"axes",
aidge_node.get_operator().axes()
))
return [onnx_node]
Loading