diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-24 13:18:32 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-24 13:22:41 -0700 |
| commit | bb1924cf6b1f3eca2e859157483b42698f41f975 (patch) | |
| tree | 4d186c3fe1e372576ef2c3f7eaa03cbbb85366ee /tensorflow/core/grappler/op_types.cc | |
| parent | 2eb7289b97d4daac044b6f0a005cc8f9fc48ac0e (diff) | |
add maxpoolgrad transposer for layout optimizer.
PiperOrigin-RevId: 197942180
Diffstat (limited to 'tensorflow/core/grappler/op_types.cc')
| -rw-r--r-- | tensorflow/core/grappler/op_types.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/op_types.cc b/tensorflow/core/grappler/op_types.cc index 6d2adcf5b8..fe0fad9148 100644 --- a/tensorflow/core/grappler/op_types.cc +++ b/tensorflow/core/grappler/op_types.cc @@ -203,6 +203,8 @@ bool IsMax(const NodeDef& node) { return node.op() == "Max"; } bool IsMaximum(const NodeDef& node) { return node.op() == "Maximum"; } +bool IsMaxPoolGrad(const NodeDef& node) { return node.op() == "MaxPoolGrad"; } + bool IsMean(const NodeDef& node) { return node.op() == "Mean"; } bool IsMerge(const NodeDef& node) { |