Home
last modified time | relevance | path

Searched refs:in_edges_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc80 const std::vector<SimpleEdge*>& in_edges() const { return in_edges_; } in in_edges()
85 res.reserve(in_edges_.size()); in in_nodes()
86 for (const auto e : in_edges_) { in in_nodes()
107 std::vector<SimpleEdge*> in_edges_; member in tensorflow::tensorrt::segment::SimpleNode
143 in_edges_.reserve(node_->in_edges().size()); in SimpleNode()
175 dst->in_edges_.push_back(edge); in SimpleGraph()
197 dst->in_edges_.push_back(edge); in AddEdge()
213 for (auto it = dst->in_edges_.begin(); it != dst->in_edges_.end(); ++it) { in RemoveEdge()
215 dst->in_edges_.erase(it); in RemoveEdge()
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc135 DCHECK(in_edges_.empty()); in Initialize()
146 in_edges_.clear(); in Clear()
197 return gtl::make_range(NeighborIter(in_edges_.begin(), true), in in_nodes()
198 NeighborIter(in_edges_.end(), true)); in in_nodes()
450 while (!node->in_edges_.empty()) { in RemoveNode()
451 RemoveEdge(*node->in_edges_.begin()); in RemoveNode()
484 CHECK(dest->in_edges_.insert(e).second); in AddEdge()
494 CHECK_EQ(e->dst_->in_edges_.erase(e), size_t{1}); in RemoveEdge()
Dgraph.h137 const EdgeSet& in_edges() const { return in_edges_; } in in_edges()
276 EdgeSet in_edges_; variable