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):
1118
1118
self ._run_test_case ([_OUTPUT ], {_INPUT : x_val })
1119
1119
tf .reset_default_graph ()
1120
1120
1121
+ @check_target ("rs6" , "onehot" )
1121
1122
def test_onehot0 (self ):
1122
1123
x_val = np .array ([0 , 1 , 2 ], dtype = np .int32 )
1123
1124
depth = 5
@@ -1138,6 +1139,7 @@ def test_onehot1(self):
1138
1139
_ = tf .identity (x_ , name = _TFOUTPUT )
1139
1140
self ._run_test_case ([_OUTPUT ], {_INPUT : x_val })
1140
1141
1142
+ @check_target ("rs6" , "onehot" )
1141
1143
def test_onehot2 (self ):
1142
1144
for axis in [- 1 , 0 , 1 ]:
1143
1145
tf .reset_default_graph ()
@@ -1148,6 +1150,7 @@ def test_onehot2(self):
1148
1150
_ = tf .identity (x_ , name = _TFOUTPUT )
1149
1151
self ._run_test_case ([_OUTPUT ], {_INPUT : x_val })
1150
1152
1153
+ @check_target ("rs6" , "onehot" )
1151
1154
@check_opset_min_version (9 , "onehot" )
1152
1155
def test_onehot3 (self ):
1153
1156
# rank 1
You can’t perform that action at this time.
0 commit comments