From 18c67a44ace913d30dc573486dc792300a2cdad3 Mon Sep 17 00:00:00 2001 From: Bixia Zheng Date: Thu, 31 May 2018 16:25:24 -0700 Subject: [PATCH] Handle FilterLayout::kOutputYXInput in FilterDescriptor::ToShortString. This fixes an error when running resnet50_batch128_fp16 with --v=2. PiperOrigin-RevId: 198798196 --- tensorflow/stream_executor/dnn.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index eed93ef..5315d1f 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -407,6 +407,8 @@ string FilterDescriptor::ToShortString() const { switch (layout_) { case FilterLayout::kOutputInputYX: return port::StrCat(od, id, spatial); + case FilterLayout::kOutputYXInput: + return port::StrCat(od, spatial, id); case FilterLayout::kOutputInputYX4: return port::StrCat(od, id, spatial, "(VECT_C)"); case FilterLayout::kInputYXOutput: -- 2.7.4