Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.h53 struct NodePtrCompare { struct
62 const std::set<const Node*, NodePtrCompare>& nodes) in Segment() argument
65 std::set<const Node*, NodePtrCompare> nodes;
Dsegment.cc937 std::set<const Node*, NodePtrCompare>& segment_nodes = itr.second.nodes; in SegmentGraph()
988 std::set<const Node*, NodePtrCompare> visited; in SegmentGraph()
989 std::set<const Node*, NodePtrCompare> logged(que->begin(), que->end()); in SegmentGraph()
1025 std::set<const Node*, NodePtrCompare> segment_nodes( in SegmentGraph()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc67 using ::tensorflow::tensorrt::segment::NodePtrCompare;
140 const std::set<const Node*, NodePtrCompare>& segment_nodes = segment.nodes; in GetEngineInfo()