diff options
| author |  Jingyue Wu <jingyue@google.com> | 2017-06-27 10:29:36 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-27 10:33:14 -0700 |
| commit | 7f0375ee24e1789eced1934e0aab4c5058486924 (patch) | |
| tree | dd98a346d7400d28ce93c51cccc8de18a4b5d0db /tensorflow/stream_executor/stream.cc | |
| parent | f062ffbbad0f578c81e44e01bb0c4df2e582ad33 (diff) | |
[SE] ThenConvolveWithAlgorithm vlogs algorithm configs.
PiperOrigin-RevId: 160292762
Diffstat (limited to 'tensorflow/stream_executor/stream.cc')
| -rw-r--r-- | tensorflow/stream_executor/stream.cc | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/tensorflow/stream_executor/stream.cc b/tensorflow/stream_executor/stream.cc index cbf8c11ef1..97193af777 100644 --- a/tensorflow/stream_executor/stream.cc +++ b/tensorflow/stream_executor/stream.cc @@ -59,6 +59,10 @@ string ToVlogString(dnn::ActivationMode mode) { return dnn::ActivationModeString(mode); } +string ToVlogString(const dnn::AlgorithmConfig &algo_config) { + return algo_config.ToString(); +} + string ToVlogString(dnn::ElementwiseOperation op) { return dnn::ElementwiseOperationString(op); } @@ -482,7 +486,8 @@ Stream &Stream::ThenConvolveWithAlgorithm( VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), PARAM(filter_descriptor), PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), - PARAM(activation_mode), PARAM(output_descriptor), PARAM(output)); + PARAM(activation_mode), PARAM(output_descriptor), PARAM(output), + PARAM(algorithm_config)); if (ok()) { if (dnn::DnnSupport *dnn = parent_->AsDnn()) { @@ -515,7 +520,8 @@ Stream &Stream::ThenConvolveWithAlgorithm( VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), PARAM(filter_descriptor), PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), - PARAM(activation_mode), PARAM(output_descriptor), PARAM(output)); + PARAM(activation_mode), PARAM(output_descriptor), PARAM(output), + PARAM(algorithm_config)); if (ok()) { if (dnn::DnnSupport *dnn = parent_->AsDnn()) { @@ -546,7 +552,7 @@ Stream &Stream::ThenConvolveWithAlgorithm( VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), PARAM(filter_descriptor), PARAM(filter_data), PARAM(convolution_descriptor), PARAM(output_descriptor), - PARAM(output)); + PARAM(output), PARAM(algorithm_config)); if (ok()) { if (dnn::DnnSupport *dnn = parent_->AsDnn()) { @@ -577,7 +583,7 @@ Stream &Stream::ThenConvolveWithAlgorithm( VLOG_CALL(PARAM(input_descriptor), PARAM(input_data), PARAM(filter_descriptor), PARAM(filter_data), PARAM(convolution_descriptor), PARAM(output_descriptor), - PARAM(output)); + PARAM(output), PARAM(algorithm_config)); if (ok()) { if (dnn::DnnSupport *dnn = parent_->AsDnn()) { |