diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-22 13:49:08 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-22 13:51:22 -0700 |
commit | 9d2c6ff2a542b9bd89b42e3b88e6299eae9bdcc4 (patch) | |
tree | 301c1026d384945565a96226d66180e1e950b3b3 /tensorflow/core/graph/graph.cc | |
parent | 4d134bad0403ebb5722144d8f859a04a5f21efc2 (diff) |
Collective Ops Part 7
Complete just enough of the core implementation to run
multi-device collectives locally within a single process.
Interfaces are still private and not availble for general use.
PiperOrigin-RevId: 197617132
Diffstat (limited to 'tensorflow/core/graph/graph.cc')
-rw-r--r-- | tensorflow/core/graph/graph.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph.cc b/tensorflow/core/graph/graph.cc index 71d0637dc2..0f748515ef 100644 --- a/tensorflow/core/graph/graph.cc +++ b/tensorflow/core/graph/graph.cc @@ -80,6 +80,9 @@ const std::unordered_map<string, Node::NodeClass>& Node::kNodeClassTable = {"Shape", NC_METADATA}, {"Rank", NC_METADATA}, {"_ScopedAllocator", NC_SCOPED_ALLOCATOR}, + {"CollectiveReduce", NC_COLLECTIVE}, + {"CollectiveBcastSend", NC_COLLECTIVE}, + {"CollectiveBcastRecv", NC_COLLECTIVE}, }); #undef REF_CLASS |