diff options
| author |  Tim Shen <timshen@google.com> | 2018-09-18 17:50:43 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-18 17:54:37 -0700 |
| commit | dff19b5a8b36ddf4aa51ce978d97b63129a7fdeb (patch) | |
| tree | 8e77b80692712648702512de0316f97e6b8b0655 | |
| parent | 38d8f893e0ab8376cf97c40fde78002f31776c92 (diff) | |
Add layout information to logging.
PiperOrigin-RevId: 213551652
| -rw-r--r-- | tensorflow/compiler/xla/service/gpu/cudnn_convolution_runner.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/cudnn_convolution_runner.cc b/tensorflow/compiler/xla/service/gpu/cudnn_convolution_runner.cc index 2a86ac265e..3310ee848e 100644 --- a/tensorflow/compiler/xla/service/gpu/cudnn_convolution_runner.cc +++ b/tensorflow/compiler/xla/service/gpu/cudnn_convolution_runner.cc @@ -92,9 +92,9 @@ Status RunCudnnConvolutionImpl(CudnnConvParams params, VLOG(3) << "tensor_ops_enabled: " << algorithm.algorithm().tensor_ops_enabled(); VLOG(3) << "Convolution kind: " << CudnnConvKindToString(kind); - VLOG(3) << "input shape: { " << ShapeUtil::HumanString(input_shape) << " }"; - VLOG(3) << "filter shape: { " << ShapeUtil::HumanString(filter_shape) << " }"; - VLOG(3) << "Output shape: { " << ShapeUtil::HumanString(output_shape) << " }"; + VLOG(3) << "input shape: " << ShapeUtil::HumanStringWithLayout(input_shape); + VLOG(3) << "filter shape: " << ShapeUtil::HumanStringWithLayout(filter_shape); + VLOG(3) << "Output shape: " << ShapeUtil::HumanStringWithLayout(output_shape); VLOG(3) << "Window: { " << window.ShortDebugString() << " }"; VLOG(3) << "Dim nums: { " << dnums.ShortDebugString() << " }"; |