Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc55 class SimpleEdge { class
57 SimpleEdge(int id, SimpleNode* src, int src_port, SimpleNode* dst, in SimpleEdge() function in tensorflow::tensorrt::segment::__anon8f933a440111::SimpleEdge
65 ~SimpleEdge() {} in ~SimpleEdge()
87 const std::vector<SimpleEdge*>& in_edges() const { return in_edges_; } in in_edges()
88 const std::vector<SimpleEdge*>& out_edges() const { return out_edges_; } in out_edges()
114 std::vector<SimpleEdge*> in_edges_;
115 std::vector<SimpleEdge*> out_edges_;
128 void RemoveEdge(const SimpleEdge*);
142 std::vector<SimpleEdge*> edges_;
178 auto edge = new SimpleEdge(i, src, e->src_output(), dst, e->dst_input(), in SimpleGraph()
[all …]