Searched refs:subgraph_nodes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 136 std::vector<const Node*> subgraph_nodes; // Topologically sorted nodes. in GetEngineInfo() local
159 subgraph_nodes.push_back(node); in GetEngineInfo()
249 subgraph_nodes.insert(subgraph_nodes.begin(), added_const_nodes.begin(), in GetEngineInfo()
253 g, graph_properties, subgraph_nodes, &info->connections, in GetEngineInfo()
|
| D | convert_nodes.h | 135 const std::vector<const Node*>& subgraph_nodes,
|
| D | convert_nodes.cc | 6456 const std::vector<const Node*>& subgraph_nodes, // In topological order in ConvertSegmentToGraphDef() argument
6536 string local_scope = subgraph_nodes.front()->name(); in ConvertSegmentToGraphDef()
6537 for (const Node* node : subgraph_nodes) { in ConvertSegmentToGraphDef()
6558 for (const Node* node : subgraph_nodes) { in ConvertSegmentToGraphDef()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 233 const std::unordered_set<string>& subgraph_nodes,
|
| D | remote_fused_graph_execute_utils.cc | 1040 const std::unordered_set<string>& subgraph_nodes, in FuseRemoteGraphByNodeNames() argument
1045 subgraph_nodes, input_graph_def, &ci_vec)); in FuseRemoteGraphByNodeNames()
|