aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-07-17 14:32:36 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-17 14:36:36 -0700
commitc9fdfab3781bd55cf3d981b98982a1e8122de8d2 (patch)
tree712fcf6ca3ab5b5410b217604ee03ec15baf5c3e /tensorflow/stream_executor
parent8c5d2127182e0fadc0dcd6e97cb4acfba3a4c343 (diff)
Print kOutputYXInput.
PiperOrigin-RevId: 204975773
Diffstat (limited to 'tensorflow/stream_executor')
-rw-r--r--tensorflow/stream_executor/dnn.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc
index 82aa8ceb32..2a30f922bc 100644
--- a/tensorflow/stream_executor/dnn.cc
+++ b/tensorflow/stream_executor/dnn.cc
@@ -117,6 +117,8 @@ string FilterLayoutString(FilterLayout layout) {
switch (layout) {
case FilterLayout::kOutputInputYX:
return "OutputInputYX";
+ case FilterLayout::kOutputYXInput:
+ return "OutputYXInput";
case FilterLayout::kOutputInputYX4:
return "OutputInputYX4";
case FilterLayout::kInputYXOutput: