diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-01 00:18:19 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-01 00:20:43 -0700 |
commit | 961a39346d8be33cff473f1e81498b887c155070 (patch) | |
tree | d1175e89f82bd60137cf9fb2ecbee64d4ac5e59c /tensorflow/stream_executor/dnn.cc | |
parent | 54b20c4be0372fb14ec9a289e4d7de7f67c03ff6 (diff) |
Unify error handling in CudnnSupport.
PiperOrigin-RevId: 198836479
Diffstat (limited to 'tensorflow/stream_executor/dnn.cc')
-rw-r--r-- | tensorflow/stream_executor/dnn.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index 5315d1f3da..82aa8ceb32 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -141,6 +141,10 @@ string PadAlignmentString(PadAlignment alignment) { return "unknown pad alignment"; } +std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment) { + return str << PadAlignmentString(alignment); +} + string ShortPoolingModeString(PoolingMode mode) { switch (mode) { case PoolingMode::kMaximum: |