From 40ebe5778e04e3c4603531c219dc1e16dace6f90 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 3 Apr 2019 16:50:55 -0700
Subject: [PATCH] Add layer name to PoolingDescriptor.

PiperOrigin-RevId: 241833294
---
 tensorflow/stream_executor/dnn.h | 6 ++++++
 1 file changed, 6 insertions(+)

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<const int64> padding() const { return padding_; }
   absl::Span<const int64> 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<int64> window_;