File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed
Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -1118,6 +1118,7 @@ def test_sign(self):
11181118 self ._run_test_case ([_OUTPUT ], {_INPUT : x_val })
11191119 tf .reset_default_graph ()
11201120
1121+ @check_target ("rs6" , "onehot" )
11211122 def test_onehot0 (self ):
11221123 x_val = np .array ([0 , 1 , 2 ], dtype = np .int32 )
11231124 depth = 5
@@ -1138,6 +1139,7 @@ def test_onehot1(self):
11381139 _ = tf .identity (x_ , name = _TFOUTPUT )
11391140 self ._run_test_case ([_OUTPUT ], {_INPUT : x_val })
11401141
1142+ @check_target ("rs6" , "onehot" )
11411143 def test_onehot2 (self ):
11421144 for axis in [- 1 , 0 , 1 ]:
11431145 tf .reset_default_graph ()
@@ -1148,6 +1150,7 @@ def test_onehot2(self):
11481150 _ = tf .identity (x_ , name = _TFOUTPUT )
11491151 self ._run_test_case ([_OUTPUT ], {_INPUT : x_val })
11501152
1153+ @check_target ("rs6" , "onehot" )
11511154 @check_opset_min_version (9 , "onehot" )
11521155 def test_onehot3 (self ):
11531156 # rank 1
You can’t perform that action at this time.
0 commit comments