| 190 | } |
| 191 | |
| 192 | string PadAlignmentString(PadAlignment alignment) { |
| 193 | switch (alignment) { |
| 194 | case PadAlignment::kDefault: |
| 195 | return "default"; |
| 196 | case PadAlignment::kCudnnPadding: |
| 197 | return "cuDNN padding"; |
| 198 | case PadAlignment::kTensorFlowPadding: |
| 199 | return "TensorFlow padding"; |
| 200 | } |
| 201 | return "unknown pad alignment"; |
| 202 | } |
| 203 | |
| 204 | std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment) { |
| 205 | return str << PadAlignmentString(alignment); |