diff options
author | Yifei Feng <yifeif@google.com> | 2017-11-21 23:32:43 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-21 23:36:22 -0800 |
commit | c6d603f02e1a98f871912cda6716cdcbed6b439e (patch) | |
tree | 4658788770c03d8a5bfd48eee59c247559344f08 /tensorflow/stream_executor/dnn.h | |
parent | f93c48dc061d23495a4425fcad17d55159cb02b1 (diff) |
Merge changes from github.
PiperOrigin-RevId: 176615107
Diffstat (limited to 'tensorflow/stream_executor/dnn.h')
-rw-r--r-- | tensorflow/stream_executor/dnn.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index 49235167ab..0d2cd4a9f2 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -661,6 +661,10 @@ class PoolingDescriptor { SetDim(&strides_, dim, value); return *this; } + PoolingDescriptor& set_propagate_nans(bool value) { + propagate_nans_ = value; + return *this; + } int ndims() const { return ndims_; } void CloneFrom(const PoolingDescriptor& other); @@ -681,10 +685,12 @@ class PoolingDescriptor { std::vector<int64> window() const { return window_; } std::vector<int64> padding() const { return padding_; } std::vector<int64> strides() const { return strides_; } + bool propagate_nans() const { return propagate_nans_; } private: PoolingMode mode_; int ndims_; + bool propagate_nans_; // Stored as: ..., y, x. std::vector<int64> window_; |