diff options
author | Xiaoqiang Zheng <zhengxq@google.com> | 2017-03-21 15:53:52 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-21 17:11:32 -0700 |
commit | be81281b3c9aba3749719e7b3b08cfb51ed55b42 (patch) | |
tree | 711475ee6227ed24f4aea9887454046434942100 /tensorflow/stream_executor/dnn.cc | |
parent | bc4bed8d08183dbb8232ee89f944e9a44e421782 (diff) |
Add a multi-pass mode for autotuning. It tests the autotune multiple times,
and only if the winning margin exceeds a certain threshold, it is accepted.
Change: 150822045
Diffstat (limited to 'tensorflow/stream_executor/dnn.cc')
-rw-r--r-- | tensorflow/stream_executor/dnn.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index 943673ae35..e834119bf8 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -194,6 +194,12 @@ std::vector<int64> ReorderDims(const std::vector<int64>& input, return reordered; } +// -- AlgorithmConfig + +string AlgorithmConfig::ToString() const { + return port::StrCat(algorithm_, ", ", algorithm_no_scratch_); +} + // -- BatchDescriptor BatchDescriptor::BatchDescriptor(int ndims) |