diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index bb3898a5fc4..34de1512ee5 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -707,6 +707,10 @@ class PoolingDescriptor { propagate_nans_ = value; return *this; } + PoolingDescriptor& set_name(const string& name) { + name_ = name; + return *this; + } int ndims() const { return ndims_; } void CloneFrom(const PoolingDescriptor& other); @@ -728,11 +732,13 @@ class PoolingDescriptor { absl::Span padding() const { return padding_; } absl::Span strides() const { return strides_; } bool propagate_nans() const { return propagate_nans_; } + string name() const { return name_; } private: PoolingMode mode_; int ndims_; bool propagate_nans_; + string name_; // Name as in Tensorflow NodeDef, for debugging purposes. // Stored as: ..., y, x. std::vector window_;