| 120 | } |
| 121 | |
| 122 | string ActivationModeString(ActivationMode mode) { |
| 123 | switch (mode) { |
| 124 | case ActivationMode::kNone: |
| 125 | return "identity"; |
| 126 | case ActivationMode::kSigmoid: |
| 127 | return "sigmoid"; |
| 128 | case ActivationMode::kRelu: |
| 129 | return "relu"; |
| 130 | case ActivationMode::kRelu6: |
| 131 | return "relu6"; |
| 132 | case ActivationMode::kReluX: |
| 133 | return "reluX"; |
| 134 | case ActivationMode::kTanh: |
| 135 | return "tanh"; |
| 136 | case ActivationMode::kBandPass: |
| 137 | return "bandpass"; |
| 138 | default: |
| 139 | LOG(FATAL) << "Unknown activation_mode " << static_cast<int32>(mode); |
| 140 | } |
| 141 | return "unknown activation_mode"; |
| 142 | } |
| 143 | |
| 144 | string ElementwiseOperationString(ElementwiseOperation op) { |
| 145 | switch (op) { |
no outgoing calls
no test coverage detected