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_grad_filter_ops.cc | |
parent | 49f147388676d77532598afb881b8a3bbf97bd41 (diff) |
Added data type info to conv autotune parameters.
Change: 150459431
Diffstat (limited to 'tensorflow/core/kernels/conv_grad_filter_ops.cc')
-rw-r--r-- | tensorflow/core/kernels/conv_grad_filter_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_grad_filter_ops.cc b/tensorflow/core/kernels/conv_grad_filter_ops.cc index af2c907cd3..80293eed82 100644 --- a/tensorflow/core/kernels/conv_grad_filter_ops.cc +++ b/tensorflow/core/kernels/conv_grad_filter_ops.cc @@ -638,6 +638,7 @@ class Conv2DSlowBackpropFilterOp : public OpKernel { "TF_CUDNN_WORKSPACE_LIMIT_IN_MB", 1LL << 32 // 4GB by default ); int device_id = stream->parent()->device_ordinal(); + DataType dtype = input.dtype(); ConvParameters conv_parameters = { dims.batch_size, // batch dims.in_depth, // in_depths @@ -650,6 +651,7 @@ class Conv2DSlowBackpropFilterOp : public OpKernel { dims.spatial_dims[1].stride}}, // stride_cols {{padding_rows, // padding_rows padding_cols}}, // padding_cols + dtype, // tensor datatype device_id, // device_id }; AlgorithmConfig algorithm_config; |