diff options
author | Ayush Dubey <ayushd@google.com> | 2018-04-30 10:36:00 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-30 10:39:01 -0700 |
commit | 09e529ff5adb916e40481563698dee72e8a15162 (patch) | |
tree | 26b37a15d699bc92717584e137999c84a592a1cb /tensorflow/core/graph/graph.cc | |
parent | a3ae05d256a9fe82d9a2e50d3f72c3361c1162e4 (diff) |
Prepare nodes that will be allocated using ScopedAllocator.
This includes changes to Executor that (1) set scope_id on nodes that are
decorated with _scoped_allocator attribute, (2) mark such nodes to never
forward input.
PiperOrigin-RevId: 194807086
Diffstat (limited to 'tensorflow/core/graph/graph.cc')
-rw-r--r-- | tensorflow/core/graph/graph.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph.cc b/tensorflow/core/graph/graph.cc index fb8a6c39e6..eeb6c60f71 100644 --- a/tensorflow/core/graph/graph.cc +++ b/tensorflow/core/graph/graph.cc @@ -79,6 +79,7 @@ const std::unordered_map<string, Node::NodeClass>& Node::kNodeClassTable = {"Size", NC_METADATA}, {"Shape", NC_METADATA}, {"Rank", NC_METADATA}, + {"_ScopedAllocator", NC_SCOPED_ALLOCATOR}, }); #undef REF_CLASS |