File tree Expand file tree Collapse file tree 6 files changed +5
-153
lines changed Expand file tree Collapse file tree 6 files changed +5
-153
lines changed Load Diff This file was deleted.
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -46,3 +46,5 @@ unary_op:
4646 OPERATOR : leaky_relu(X, A)
4747 - NAME : round
4848 OPERATOR : round(X)
49+ - NAME : tan
50+ OPERATOR : tan(X)
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -154,6 +154,7 @@ DEFINE_ACTIVATION_FN(hardswish);
154154DEFINE_ACTIVATION_FN (hardsigmoid);
155155DEFINE_LEAKY_RELU_FN (leaky_relu);
156156DEFINE_ACTIVATION_FN (round);
157+ DEFINE_ACTIVATION_FN (tan);
157158
158159REGISTER_OPERATORS {
159160 VK_REGISTER_OP (aten.abs .default , abs);
@@ -174,6 +175,7 @@ REGISTER_OPERATORS {
174175 VK_REGISTER_OP (aten.hardsigmoid .default , hardsigmoid);
175176 VK_REGISTER_OP (aten.leaky_relu .default , leaky_relu);
176177 VK_REGISTER_OP (aten.round .default , round);
178+ VK_REGISTER_OP (aten.tan .default , tan);
177179}
178180
179181} // namespace vkcompute
Original file line number Diff line number Diff line change @@ -1147,6 +1147,7 @@ def get_reduce_op_inputs():
11471147 "aten.hardsigmoid.default" ,
11481148 "aten.leaky_relu.default" ,
11491149 "aten.round.default" ,
1150+ "aten.tan.default" ,
11501151 ]
11511152)
11521153def get_unary_ops_inputs ():
@@ -1164,22 +1165,6 @@ def get_unary_ops_inputs():
11641165 return test_suite
11651166
11661167
1167- # separate test suite from unary_ops for learning purposes
1168- @register_test_suite ("aten.tan.default" )
1169- def get_tan_inputs ():
1170- test_suite = VkTestSuite (
1171- [
1172- (M1 ,),
1173- (M1 , M2 ),
1174- (S1 , M1 , M2 ),
1175- (S1 , S2 , S2 , M2 ),
1176- ]
1177- )
1178- test_suite .storage_types = ["utils::kTexture3D" , "utils::kBuffer" ]
1179- test_suite .dtypes = ["at::kFloat" ]
1180- return test_suite
1181-
1182-
11831168@register_test_suite ("aten._native_batch_norm_legit_no_training.default" )
11841169def get_native_batch_norm_inputs ():
11851170 Test = namedtuple (
You can’t perform that action at this time.
0 commit comments