add OP sigmoid (#12056)

Co-authored-by: Judd <foldl@boxvest.com>
This commit is contained in:
Judd
2025-02-25 19:32:20 +08:00
committed by GitHub
parent 393fca629e
commit c132239bfb
3 changed files with 35 additions and 0 deletions

View File

@ -482,6 +482,7 @@ void process_shaders() {
string_to_spv("relu_f32", "relu.comp", {{"A_TYPE", "float"}, {"D_TYPE", "float"}});
string_to_spv("leaky_relu_f32", "leaky_relu.comp", {{"A_TYPE", "float"}, {"D_TYPE", "float"}});
string_to_spv("tanh_f32", "tanh.comp", {{"A_TYPE", "float"}, {"D_TYPE", "float"}});
string_to_spv("sigmoid_f32", "sigmoid.comp", {{"A_TYPE", "float"}, {"D_TYPE", "float"}});
string_to_spv("diag_mask_inf_f32", "diag_mask_inf.comp", {{"A_TYPE", "float"}, {"D_TYPE", "float"}});