diff options
| author |  Yangzihao Wang <yangzihao@google.com> | 2017-03-17 10:01:54 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-17 11:24:43 -0700 |
| commit | f4b237f8cdd25a45dc26adc61c3086c2575f5396 (patch) | |
| tree | b772ab8810b4406d42cc6105d8de7deb99902380 /tensorflow/core/kernels/conv_ops.cc | |
| parent | 49f147388676d77532598afb881b8a3bbf97bd41 (diff) | |
Added data type info to conv autotune parameters.
Change: 150459431
Diffstat (limited to 'tensorflow/core/kernels/conv_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/conv_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_ops.cc b/tensorflow/core/kernels/conv_ops.cc index 9dd7bdb786..fd7c825cce 100644 --- a/tensorflow/core/kernels/conv_ops.cc +++ b/tensorflow/core/kernels/conv_ops.cc @@ -636,6 +636,7 @@ void LaunchConv2DOp<GPUDevice, T>::launch( ); int device_id = stream->parent()->device_ordinal(); + DataType dtype = input.dtype(); ConvParameters conv_parameters = { in_batch, // batch in_depths, // in_depths @@ -648,6 +649,7 @@ void LaunchConv2DOp<GPUDevice, T>::launch( col_stride}}, // stride_cols {{padding_rows, // padding_rows padding_cols}}, // padding_cols + dtype, // tensor datatype device_id, // device_id }; AlgorithmConfig algorithm_config; |