HomeSort by relevance Sort by last modified time
    Searched refs:subgraph_node_ids (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/convert/
convert_nodes.h 36 const tensorflow::Graph& graph, const std::set<int>& subgraph_node_ids,
convert_graph.cc 64 const std::set<int>& subgraph_node_ids,
66 for (int node_id : subgraph_node_ids) {
69 if (!subgraph_node_ids.count(edge->src()->id()) &&
78 const std::set<int>& subgraph_node_ids,
80 for (int node_id : subgraph_node_ids) {
83 if (!subgraph_node_ids.count(edge->dst()->id()) &&
115 const std::set<int>& subgraph_node_ids,
122 GetSubGraphIncomingEdges(*graph, subgraph_node_ids, &subgraph_incoming_edges);
133 for (int node_id : subgraph_node_ids) {
143 GetSubGraphOutgoingEdges(*graph, subgraph_node_ids, &subgraph_outgoing_edges)
256 std::set<int> subgraph_node_ids; local
    [all...]
convert_nodes.cc     [all...]

Completed in 103 milliseconds