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.cc87 const std::vector<SimpleEdge*>& in_edges() const { return in_edges_; } in in_edges()
92 res.reserve(in_edges_.size()); in in_nodes()
93 for (const auto e : in_edges_) { in in_nodes()
114 std::vector<SimpleEdge*> in_edges_; member in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleNode
150 in_edges_.reserve(node_->in_edges().size()); in SimpleNode()
182 dst->in_edges_.push_back(edge); in SimpleGraph()
204 dst->in_edges_.push_back(edge); in AddEdge()
220 for (auto it = dst->in_edges_.begin(); it != dst->in_edges_.end(); ++it) { in RemoveEdge()
222 dst->in_edges_.erase(it); in RemoveEdge()
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc133 DCHECK(in_edges_.empty()); in Initialize()
143 in_edges_.clear(); in Clear()
203 return gtl::make_range(NeighborIter(in_edges_.begin(), true), in in_nodes()
204 NeighborIter(in_edges_.end(), true)); in in_nodes()
434 copy->in_edges_.reserve(n->in_edges().size()); in Copy()
499 for (const Edge* e : node->in_edges_) { in RemoveNode()
505 node->in_edges_.clear(); in RemoveNode()
507 CHECK_EQ(e->dst_->in_edges_.erase(e), size_t{1}); in RemoveNode()
541 CHECK(dest->in_edges_.insert(e).second); in AddEdge()
551 CHECK_EQ(e->dst_->in_edges_.erase(e), size_t{1}); in RemoveEdge()
Dgraph.h147 const EdgeSet& in_edges() const { return in_edges_; } in in_edges()
323 EdgeSet in_edges_; variable