| 22 | } |
| 23 | |
| 24 | Mode string_to_mode(std::string mode_str) { |
| 25 | if ("RELU" == mode_str) { |
| 26 | return Mode::RELU; |
| 27 | } else if ("EXP" == mode_str) { |
| 28 | return Mode::EXP; |
| 29 | } else if ("ABS" == mode_str) { |
| 30 | return Mode::ABS; |
| 31 | } else if ("NEGATE" == mode_str) { |
| 32 | return Mode::NEGATE; |
| 33 | } else if ("H_SWISH" == mode_str) { |
| 34 | return Mode::H_SWISH; |
| 35 | } else if ("ADD" == mode_str) { |
| 36 | return Mode::ADD; |
| 37 | } else if ("SUB" == mode_str) { |
| 38 | return Mode::SUB; |
| 39 | } else if ("MUL" == mode_str) { |
| 40 | return Mode::MUL; |
| 41 | } else if ("MAX" == mode_str) { |
| 42 | return Mode::MAX; |
| 43 | } else if ("MIN" == mode_str) { |
| 44 | return Mode::MIN; |
| 45 | } else if ("SIGMOID" == mode_str) { |
| 46 | return Mode::SIGMOID; |
| 47 | } else if ("TRUE_DIV" == mode_str) { |
| 48 | return Mode::TRUE_DIV; |
| 49 | } else if ("FUSE_ADD_RELU" == mode_str) { |
| 50 | return Mode::FUSE_ADD_RELU; |
| 51 | } else if ("FUSE_ADD_SIGMOID" == mode_str) { |
| 52 | return Mode::FUSE_ADD_SIGMOID; |
| 53 | } else if ("FUSE_MUL_ADD3" == mode_str) { |
| 54 | return Mode::FUSE_MUL_ADD3; |
| 55 | } else if ("FUSE_MUL_ADD4" == mode_str) { |
| 56 | return Mode::FUSE_MUL_ADD4; |
| 57 | } |
| 58 | printf("mode is not support %s\n", mode_str.c_str()); |
| 59 | } |
| 60 | |
| 61 | std::shared_ptr<TensorNDArray> fused_elemwise_compute_dnn_truth( |
| 62 | std::shared_ptr<TensorNDArray> inputs, std::vector<std::string> modes, |
no outgoing calls
no test coverage detected