Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 1 | // Copyright 2020 Google LLC |
| 2 | // |
| 3 | // This source code is licensed under the BSD-style license found in the |
| 4 | // LICENSE file in the root directory of this source tree. |
| 5 | |
| 6 | #include <math.h> |
| 7 | #include <stddef.h> |
| 8 | #include <stdint.h> |
| 9 | |
| 10 | #include <xnnpack/common.h> |
| 11 | #include <xnnpack/log.h> |
| 12 | #include <xnnpack/subgraph.h> |
| 13 | |
| 14 | |
Marat Dukhan | 210e35d | 2020-06-04 02:45:57 -0700 | [diff] [blame] | 15 | // This function is defined inline when logging is disabled |
| 16 | #if XNN_LOG_LEVEL > 0 |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 17 | const char* xnn_node_type_to_string(enum xnn_node_type type) { |
| 18 | switch (type) { |
| 19 | case xnn_node_type_invalid: |
| 20 | return "Invalid"; |
Marat Dukhan | 5fab409 | 2020-06-10 01:28:28 -0700 | [diff] [blame] | 21 | case xnn_node_type_abs: |
| 22 | return "Abs"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 23 | case xnn_node_type_add2: |
| 24 | return "Add2"; |
| 25 | case xnn_node_type_argmax_pooling_2d: |
| 26 | return "ArgMax Pooling 2D"; |
| 27 | case xnn_node_type_average_pooling_2d: |
| 28 | return "Average Pooling 2D"; |
Marat Dukhan | 5fab409 | 2020-06-10 01:28:28 -0700 | [diff] [blame] | 29 | case xnn_node_type_bankers_rounding: |
| 30 | return "Bankers Rounding"; |
| 31 | case xnn_node_type_ceiling: |
| 32 | return "Ceiling"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 33 | case xnn_node_type_clamp: |
| 34 | return "Clamp"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 35 | case xnn_node_type_convolution_2d: |
| 36 | return "Convolution 2D"; |
| 37 | case xnn_node_type_deconvolution_2d: |
| 38 | return "Deconvolution 2D"; |
| 39 | case xnn_node_type_depthwise_convolution_2d: |
| 40 | return "Depthwise Convolution 2D"; |
Artsiom Ablavatski | bbe8506 | 2020-11-05 14:07:37 -0800 | [diff] [blame] | 41 | case xnn_node_type_depth_to_space: |
Marat Dukhan | bb781b6 | 2020-11-12 13:34:05 -0800 | [diff] [blame] | 42 | return "Depth To Space"; |
Marat Dukhan | 9d3a459 | 2020-06-05 16:52:42 -0700 | [diff] [blame] | 43 | case xnn_node_type_divide: |
| 44 | return "Divide"; |
Marat Dukhan | a160020 | 2020-12-01 22:17:16 -0800 | [diff] [blame] | 45 | case xnn_node_type_elu: |
| 46 | return "ELU"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 47 | case xnn_node_type_fully_connected: |
| 48 | return "Fully Connected"; |
Marat Dukhan | 5fab409 | 2020-06-10 01:28:28 -0700 | [diff] [blame] | 49 | case xnn_node_type_floor: |
| 50 | return "Floor"; |
Marat Dukhan | a059b7d | 2020-06-11 11:41:27 -0700 | [diff] [blame] | 51 | case xnn_node_type_global_average_pooling_2d: |
| 52 | return "Global Average Pooling 2D"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 53 | case xnn_node_type_hardswish: |
| 54 | return "HardSwish"; |
Marat Dukhan | 5bbebac | 2020-06-10 19:42:15 -0700 | [diff] [blame] | 55 | case xnn_node_type_leaky_relu: |
| 56 | return "Leaky ReLU"; |
Marat Dukhan | 9d3a459 | 2020-06-05 16:52:42 -0700 | [diff] [blame] | 57 | case xnn_node_type_maximum2: |
| 58 | return "Maximum2"; |
| 59 | case xnn_node_type_minimum2: |
| 60 | return "Minimum2"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 61 | case xnn_node_type_multiply2: |
| 62 | return "Multiply2"; |
| 63 | case xnn_node_type_max_pooling_2d: |
| 64 | return "Max Pooling 2D"; |
Marat Dukhan | 5fab409 | 2020-06-10 01:28:28 -0700 | [diff] [blame] | 65 | case xnn_node_type_negate: |
| 66 | return "Negate"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 67 | case xnn_node_type_prelu: |
| 68 | return "PReLU"; |
| 69 | case xnn_node_type_sigmoid: |
| 70 | return "Sigmoid"; |
| 71 | case xnn_node_type_softmax: |
| 72 | return "Softmax"; |
Marat Dukhan | aff24e2 | 2020-07-23 01:43:58 -0700 | [diff] [blame] | 73 | case xnn_node_type_static_constant_pad: |
| 74 | return "Static Constant Pad"; |
Marat Dukhan | d27202d | 2020-07-09 23:43:40 -0700 | [diff] [blame] | 75 | case xnn_node_type_static_reshape: |
| 76 | return "Static Reshape"; |
Marat Dukhan | aff24e2 | 2020-07-23 01:43:58 -0700 | [diff] [blame] | 77 | case xnn_node_type_static_resize_bilinear_2d: |
| 78 | return "Static Resize Bilinear 2D"; |
Marat Dukhan | 5fab409 | 2020-06-10 01:28:28 -0700 | [diff] [blame] | 79 | case xnn_node_type_square: |
| 80 | return "Square"; |
Marat Dukhan | 51a01c6 | 2020-07-09 03:26:57 -0700 | [diff] [blame] | 81 | case xnn_node_type_square_root: |
| 82 | return "Square Root"; |
Marat Dukhan | 9d3a459 | 2020-06-05 16:52:42 -0700 | [diff] [blame] | 83 | case xnn_node_type_squared_difference: |
| 84 | return "Squared Difference"; |
| 85 | case xnn_node_type_subtract: |
| 86 | return "Subtract"; |
Marat Dukhan | 3b59de2 | 2020-06-03 20:15:19 -0700 | [diff] [blame] | 87 | case xnn_node_type_unpooling_2d: |
| 88 | return "Unpooling 2D"; |
| 89 | } |
| 90 | XNN_UNREACHABLE; |
| 91 | return NULL; |
| 92 | } |
Marat Dukhan | 210e35d | 2020-06-04 02:45:57 -0700 | [diff] [blame] | 93 | #endif // XNN_LOG_LEVEL > 0 |