diff options
| author |  Bixia Zheng <bixia@google.com> | 2018-05-31 16:25:24 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-31 16:29:35 -0700 |
| commit | 18c67a44ace913d30dc573486dc792300a2cdad3 (patch) | |
| tree | 83fe6b4b2bcffe0c2119e344f73d484dc2780e92 /tensorflow/stream_executor | |
| parent | 52dbe0647fbcd2c4abd5492e04414cc4169f688a (diff) | |
Handle FilterLayout::kOutputYXInput in FilterDescriptor::ToShortString.
This fixes an error when running resnet50_batch128_fp16 with --v=2.
PiperOrigin-RevId: 198798196
Diffstat (limited to 'tensorflow/stream_executor')
| -rw-r--r-- | tensorflow/stream_executor/dnn.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index eed93efc8d..5315d1f3da 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: |