Searched refs:node_input (Results 1 – 6 of 6) sorted by relevance
208 string node_input = it->second->input(i); in AddGraph() local212 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()
258 for (const GraphTransferInfo::NodeInput& node_input : in SortParams() local259 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() local766 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() local877 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 …]
178 GraphTransferInfo::NodeInput& node_input = in SetupGraph() local180 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() local215 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()
247 for (const GraphTransferInfo::NodeInput& node_input : in SanityCheckNodes() local248 input_params->node_input()) { in SanityCheckNodes()249 EXPECT_GE(node_input.output_port(), 0); in SanityCheckNodes()
44 for (const string& node_input : node.input()) { in AddNodeToGraph() local47 if (StrContains(node_input, cluster_name) && in AddNodeToGraph()55 new_node->add_input(node_input); in AddNodeToGraph()
41 repeated NodeInput node_input = 2; field