Commit9ccab70

committed

fix code

1 parent5316964 commit9ccab70Copy full SHA for 9ccab70

File tree

+15

-7

lines changed

+15

-7

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -406,7 +406,7 @@ def _run(cls, code):`
`406`	`406`	`importonnx.helper`
`407`	`407`	`importonnx.numpy_helper`
`408`	`408`	`importonnx_array_api.translate_api.make_helper`
`409`		`-importonnx.reference.custom_element_types`
	`409`	`+importml_dtypes`
`410`	`410`
`411`	`411`	`deffrom_array_extended(tensor,name=None):`
`412`	`412`	`dt=tensor.dtype`
`@@ -433,7 +433,7 @@ def from_array_extended(tensor, name=None):`
`433`	`433`	`globs.update(onnx.helper.__dict__)`
`434`	`434`	`globs.update(onnx.numpy_helper.__dict__)`
`435`	`435`	`globs.update(onnx_array_api.translate_api.make_helper.__dict__)`
`436`		`-globs.update(onnx.reference.custom_element_types.__dict__)`
	`436`	`+globs.update(ml_dtypes.__dict__)`
`437`	`437`	`globs["from_array_extended"]=from_array_extended`
`438`	`438`	`locs= {}`
`439`	`439`	`try:`

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -129,8 +129,9 @@ def render_attribute_value(self, value: Any) -> Tuple[List[str], str]:`
`129`	`129`	`ifvalue[0].type==AttributeProto.TENSOR:`
`130`	`130`	`repl= {"bool":"bool_","object":"object_","str":"str_"}`
`131`	`131`	`sdtype=repl.get(str(v.dtype),str(str(v.dtype)))`
	`132`	`+package="np"ifhasattr(np,sdtype)else"ml_dtypes"`
`132`	`133`	`return [], (`
`133`		`-f"from_array(np.array({v.tolist()}, dtype=np.{sdtype}), "`
	`134`	`+f"from_array(np.array({v.tolist()}, dtype={package}.{sdtype}), "`
`134`	`135`	`f"name={value[0].name!r})"`
`135`	`136`	`)`
`136`	`137`	`ifisinstance(v, (int,float,list)):`

Lines changed: 5 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`fromtypingimportAny,Dict,List`
	`2`	`+importnumpyasnp`
`2`	`3`	`fromonnximportTensorProto`
`3`	`4`	`fromonnx.numpy_helperimportto_array`
`4`	`5`	`from .base_emitterimportBaseEmitter`
`@@ -135,7 +136,10 @@ def _emit_end_signature(self, **kwargs: Dict[str, Any]) -> List[str]:`
`135`	`136`	`val=to_array(init)`
`136`	`137`	`stype=str(val.dtype).split(".")[-1]`
`137`	`138`	`name=self._clean_result_name(init.name)`
`138`		`-rows.append(f"{name} = np.array({val.tolist()}, dtype=np.{stype})")`
	`139`	`+package="np"ifhasattr(np,stype)else"ml_dtypes"`
	`140`	`+rows.append(`
	`141`	`+f"{name} = np.array({val.tolist()}, dtype={package}.{stype})"`
	`142`	`+ )`
`139`	`143`	`returnrows`
`140`	`144`
`141`	`145`	`def_emit_begin_return(self,**kwargs:Dict[str,Any])->List[str]:`

Lines changed: 3 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`fromtypingimportAny,Dict,List,Optional,Tuple`
	`2`	`+importnumpyasnp`
`2`	`3`	`fromonnximportAttributeProto`
`3`	`4`	`from ..annotationsimportELEMENT_TYPE_NAME`
`4`	`5`	`from .base_emitterimportBaseEmitter`
`@@ -105,7 +106,7 @@ def _emit_initializer(self, **kwargs: Dict[str, Any]) -> List[str]:`
`105`	`106`	`else:`
`106`	`107`	`raiseNotImplementedError(f"Unexpected dtype={sdtype}.")`
`107`	`108`	`else:`
`108`		`-sdtype=f"np.{sdtype}"`
	`109`	`+sdtype=f"np.{sdtype}"ifhasattr(np,sdtype)elsef"ml_dtypes.{sdtype}"`
`109`	`110`
`110`	`111`	`return [`
`111`	`112`	`"initializers.append(",`
`@@ -233,7 +234,7 @@ def _emit_initializer(self, **kwargs: Dict[str, Any]) -> List[str]:`
`233`	`234`	`else:`
`234`	`235`	`raiseNotImplementedError(f"Unexpected dtype={sdtype}.")`
`235`	`236`	`else:`
`236`		`-sdtype=f"np.{sdtype}"`
	`237`	`+sdtype=f"np.{sdtype}"ifhasattr(np,sdtype)elsef"ml_dtypes.{sdtype}"`
`237`	`238`	`ifvalue.size<=16:`
`238`	`239`	`return [`
`239`	`240`	`"initializers.append(",`

Lines changed: 3 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`fromtypingimportAny,Dict,List`
	`2`	`+importnumpyasnp`
`2`	`3`	`from ..annotationsimportELEMENT_TYPE_NAME`
`3`	`4`	`from .base_emitterimportBaseEmitter`
`4`	`5`
`@@ -43,8 +44,9 @@ def _emit_initializer(self, **kwargs: Dict[str, Any]) -> List[str]:`
`43`	`44`	`value=kwargs["value"]`
`44`	`45`	`repl= {"bool":"bool_","object":"object_","str":"str_"}`
`45`	`46`	`sdtype=repl.get(str(value.dtype),str(str(value.dtype)))`
	`47`	`+package="np"ifhasattr(np,sdtype)else"ml_dtypes"`
`46`	`48`	`return [`
`47`		`-f"cst(np.array({value.tolist()}, dtype=np.{sdtype}))",`
	`49`	`+f"cst(np.array({value.tolist()}, dtype={package}.{sdtype}))",`
`48`	`50`	`f"rename({name!r})",`
`49`	`51`	`]`
`50`	`52`

Comments

(0)