Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc48 class SimpleEdge { class
50 SimpleEdge(int id, SimpleNode* src, int src_port, SimpleNode* dst, in SimpleEdge() function in tensorflow::tensorrt::segment::SimpleEdge
58 ~SimpleEdge() {} in ~SimpleEdge()
80 const std::vector<SimpleEdge*>& in_edges() const { return in_edges_; } in in_edges()
81 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges()
107 std::vector<SimpleEdge*> in_edges_;
108 std::vector<SimpleEdge*> out_edges_;
121 void RemoveEdge(const SimpleEdge*);
135 std::vector<SimpleEdge*> edges_;
171 auto edge = new SimpleEdge(i, src, e->src_output(), dst, e->dst_input(), in SimpleGraph()
[all …]