Commitb73a0cb

committed

tiny changes

1 parent1d62c51 commitb73a0cbCopy full SHA for b73a0cb

File tree

+21

-0

lines changed

+21

-0

lines changed

7.76 KB

Binary file not shown.

Lines changed: 21 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+importos`
`1`	`2`	`importunittest`
`2`	`3`	`importonnx`
`3`	`4`	`fromonnx_array_api.ext_test_caseimportExtTestCase`
`@@ -53,6 +54,26 @@ def test_keep_unused_outputs(self):`
`53`	`54`	`self.assertEqual(len(onx.graph.node),2)`
`54`	`55`	`self.assertEqual(onx.graph.node[0].op_type,"Split")`
`55`	`56`
	`57`	`+deftest_check_files(self):`
	`58`	`+importonnxruntime`
	`59`	`+`
	`60`	`+data=os.path.join(os.path.dirname(__file__),"data")`
	`61`	`+filename= [fforfinos.listdir(data)iff.endswith(".onnx")]`
	`62`	`+forfinfilename:`
	`63`	`+withself.subTest(f=f):`
	`64`	`+onx=onnx.load(os.path.join(data,f))`
	`65`	`+sess=onnxruntime.InferenceSession(`
	`66`	`+os.path.join(data,f),providers=["CPUExecutionProvider"]`
	`67`	`+ )`
	`68`	`+assertsess`
	`69`	`+g=GraphBuilder(onx)`
	`70`	`+g.optimize()`
	`71`	`+onx2=g.to_onnx()`
	`72`	`+sess2=onnxruntime.InferenceSession(`
	`73`	`+onx2.SerializeToString(),providers=["CPUExecutionProvider"]`
	`74`	`+ )`
	`75`	`+assertsess2`
	`76`	`+`
`56`	`77`
`57`	`78`	`if__name__=="__main__":`
`58`	`79`	`unittest.main(verbosity=2)`

Comments

(0)