Searched refs:subgraph_nodes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 201 std::vector<const Node*> subgraph_nodes; // Topologically sorted nodes. in GetEngineInfo() local
240 subgraph_nodes.push_back(node); in GetEngineInfo()
332 subgraph_nodes.insert(subgraph_nodes.begin(), added_const_nodes.begin(), in GetEngineInfo()
336 g, graph_properties, subgraph_nodes, &info->connections, in GetEngineInfo()
|
| D | convert_nodes.h | 137 const std::vector<const Node*>& subgraph_nodes,
|
| D | convert_nodes.cc | 4170 const std::vector<const Node*>& subgraph_nodes, // In topological order in ConvertSegmentToGraphDef() argument
4246 string local_scope = subgraph_nodes.front()->name(); in ConvertSegmentToGraphDef()
4247 for (const Node* node : subgraph_nodes) { in ConvertSegmentToGraphDef()
4268 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 | 1038 const std::unordered_set<string>& subgraph_nodes, in FuseRemoteGraphByNodeNames() argument
1043 subgraph_nodes, input_graph_def, &ci_vec)); in FuseRemoteGraphByNodeNames()
|