Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc208 string node_input = it->second->input(i); in AddGraph() local
212 auto prefix_pos = node_input.find(":"); in AddGraph()
213 if (prefix_pos != node_input.npos) { in AddGraph()
214 std::vector<string> input_parts = str_util::Split(node_input, ":"); in AddGraph()
216 << "Unknown NodeDef.input format: " << node_input; in AddGraph()
217 node_input = input_parts[0]; in AddGraph()
221 if (node_input.substr(0, 1) == "^") { in AddGraph()
222 node_input = node_input.substr(1); in AddGraph()
227 node->AddInput(node_input, output_idx, i); in AddGraph()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc258 for (const GraphTransferInfo::NodeInput& node_input : in SortParams() local
259 input_map.at(node_id)->node_input()) { in SortParams()
260 dependency_map.at(node_id).emplace(node_input.node_id()); in SortParams()
764 GraphTransferInfo::NodeInput& node_input = in RegisterPadNode() local
766 node_input.set_node_id(id); in RegisterPadNode()
767 node_input.set_output_port(0); in RegisterPadNode()
876 GraphTransferInfo::NodeInput& node_input = *node_input_info->add_node_input(); in AddNodeInputByInputIndex() local
877 node_input.set_node_id(src_id); in AddNodeInputByInputIndex()
878 node_input.set_output_port(port); in AddNodeInputByInputIndex()
892 GraphTransferInfo::NodeInput& node_input = in AppendNodeInputParams() local
[all …]
Dhexagon_control_wrapper.cc178 GraphTransferInfo::NodeInput& node_input = in SetupGraph() local
180 node_input.set_node_id(node_info->node_id()); in SetupGraph()
181 node_input.set_output_port(port); in SetupGraph()
214 const GraphTransferInfo::NodeInput& node_input = in SetupGraph() local
215 input_params.node_input(i); in SetupGraph()
216 node_ids[i] = node_input.node_id() + NODE_ID_OFFSET; in SetupGraph()
217 ports[i] = node_input.output_port(); in SetupGraph()
Dgraph_transferer_test.cc247 for (const GraphTransferInfo::NodeInput& node_input : in SanityCheckNodes() local
248 input_params->node_input()) { in SanityCheckNodes()
249 EXPECT_GE(node_input.output_port(), 0); in SanityCheckNodes()
/external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/
Dresolve_cluster.cc44 for (const string& node_input : node.input()) { in AddNodeToGraph() local
47 if (StrContains(node_input, cluster_name) && in AddNodeToGraph()
55 new_node->add_input(node_input); in AddNodeToGraph()
/external/tensorflow/tensorflow/core/framework/
Dgraph_transfer_info.proto41 repeated NodeInput node_input = 2; field