diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-11 10:43:30 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-11 10:48:25 -0700 |
| commit | 227eee585118e942e5fefa8f949562749c482f7a (patch) | |
| tree | 42782cee75a1455c612f65ea135f2e6108b3634e /tensorflow/core/grappler/op_types.h | |
| parent | b125f6ad1f94be7541d56e6edf9235b3cf68f76e (diff) | |
Use Identity instead of Snapshot when the graph does not contain ops that modify their inputs.
PiperOrigin-RevId: 196275133
Diffstat (limited to 'tensorflow/core/grappler/op_types.h')
| -rw-r--r-- | tensorflow/core/grappler/op_types.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/op_types.h b/tensorflow/core/grappler/op_types.h index f6105d710e..a5599eb22e 100644 --- a/tensorflow/core/grappler/op_types.h +++ b/tensorflow/core/grappler/op_types.h @@ -166,6 +166,10 @@ bool IsPersistent(const NodeDef& node); bool IsFreeOfSideEffect(const NodeDef& node); +// Returns true if the takes a tensor reference as input, or if looking up its +// OpDef failed. +bool MaybeHasRefInput(const NodeDef& node); + bool ModifiesFrameInfo(const NodeDef& node); // Returns true if the op is known to write to one or more of its inputs. |