experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add layer name to PoolingDescriptor.

Browse Source 
PiperOrigin-RevId: 241833294
This commit is contained in:
A. Unique TensorFlower 2019-04-03 16:50:55 -07:00 committed by TensorFlower Gardener
parent 5344006fdc
commit 40ebe5778e
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/stream_executor/dnn.h
View File

@ -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_;