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_3d.cc | |
parent | 49f147388676d77532598afb881b8a3bbf97bd41 (diff) |
Added data type info to conv autotune parameters.
Change: 150459431
Diffstat (limited to 'tensorflow/core/kernels/conv_ops_3d.cc')
-rw-r--r-- | tensorflow/core/kernels/conv_ops_3d.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_ops_3d.cc b/tensorflow/core/kernels/conv_ops_3d.cc index 9018343e25..40fde64570 100644 --- a/tensorflow/core/kernels/conv_ops_3d.cc +++ b/tensorflow/core/kernels/conv_ops_3d.cc @@ -343,6 +343,7 @@ struct LaunchConvOp<GPUDevice, T> { "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 = { in_batch, in_depth, @@ -351,6 +352,7 @@ struct LaunchConvOp<GPUDevice, T> { {{filter_planes, filter_rows, filter_cols}}, {{strides[0], strides[1], strides[2]}}, {{pad_planes, pad_rows, pad_cols}}, + dtype, device_id, }; |