Skip to content

Commit 0ba5680

Browse files
committed
[Core] add get_opset_imports utility fxn to ModelWrapper
1 parent 9f56661 commit 0ba5680

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

src/qonnx/core/modelwrapper.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -738,3 +738,7 @@ def set_tensor_sparsity(self, tensor_name, sparsity_dict):
738738
qa.tensor_name = tensor_name
739739
qa.quant_parameter_tensor_names.append(dt)
740740
qnt_annotations.append(qa)
741+
742+
def get_opset_imports(self):
743+
"""Returns a list of imported opsets as (domain, version) tuples."""
744+
return [(opset.domain, opset.version) for opset in self._model_proto.opset_import]

tests/core/test_modelwrapper.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@ def test_modelwrapper():
6868
inp_sparsity = {"dw": {"kernel_shape": [3, 3]}}
6969
model.set_tensor_sparsity(first_conv_iname, inp_sparsity)
7070
assert model.get_tensor_sparsity(first_conv_iname) == inp_sparsity
71+
assert model.get_opset_imports() == [("", 8)]
7172

7273

7374
def test_modelwrapper_set_get_rm_initializer():

0 commit comments

Comments
 (0)