Commitcb27cf5

committed

fixes for ir_version

1 parent492b6d4 commitcb27cf5Copy full SHA for cb27cf5

File tree

-0

lines changed

-0

lines changed

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -156,6 +156,7 @@ def __init__(`
`156`	`156`	`optimization_options:Optional[OptimizationOptions]=None,`
`157`	`157`	`args:Optional[List[Any]]=None,`
`158`	`158`	`verbose:int=0,`
	`159`	`+ir_version:Optional[int]=None,`
`159`	`160`	`):`
`160`	`161`	`self.optimization_options=optimization_optionsorOptimizationOptions()`
`161`	`162`	`self.as_function=as_function`
`@@ -170,6 +171,7 @@ def __init__(`
`170`	`171`	`ifisinstance(target_opset_or_existing_proto,int)`
`171`	`172`	`elsetarget_opset_or_existing_proto`
`172`	`173`	`)`
	`174`	`+self.ir_version=ir_version`
`173`	`175`	`self.nodes= []`
`174`	`176`	`self.initializers_dict= {}`
`175`	`177`	`self.inputs= []`
`@@ -186,6 +188,7 @@ def __init__(`
`186`	`188`	`),"input_names must be empty if the input is an existing model."`
`187`	`189`	`proto=target_opset_or_existing_proto`
`188`	`190`	`self.opsets= {d.domain:d.versionfordinproto.opset_import}`
	`191`	`+self.ir_version=ir_versionortarget_opset_or_existing_proto.ir_version`
`189`	`192`	`self.nodes=list(proto.graph.node)`
`190`	`193`	`self.initializers_dict= {i.name:iforiinproto.graph.initializer}`
`191`	`194`	`self.initializers_dict.update(`
`@@ -674,6 +677,8 @@ def to_onnx(`
`674`	`677`	`ifself.verbose:`
`675`	`678`	`print("[GraphBuilder] onh.make_model")`
`676`	`679`	`model=oh.make_model(graph,opset_imports=opsets)`
	`680`	`+ifself.ir_version:`
	`681`	`+model.ir_version=self.ir_version`
`677`	`682`	`returnmodel`
`678`	`683`
`679`	`684`	`def_check_order_node(self,ind:int,node:NodeProto,existing:Set[str]):`

Comments

(0)