Searched refs:node_input (Results 1 – 10 of 10) 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()
266 for (const GraphTransferNodeInput& node_input : in SortParams() local267 input_map.at(node_id)->node_input()) { in SortParams()268 dependency_map.at(node_id).emplace(node_input.node_id()); in SortParams()771 GraphTransferNodeInput& node_input = *node_input_info.add_node_input(); in RegisterPadNode() local772 node_input.set_node_id(id); in RegisterPadNode()773 node_input.set_output_port(0); in RegisterPadNode()881 GraphTransferNodeInput& node_input = *node_input_info->add_node_input(); in AddNodeInputByInputIndex() local882 node_input.set_node_id(src_id); in AddNodeInputByInputIndex()883 node_input.set_output_port(port); in AddNodeInputByInputIndex()897 GraphTransferNodeInput& node_input = *node_input_info.add_node_input(); in AppendNodeInputParams() local[all …]
180 GraphTransferNodeInput& node_input = *node_input_info.add_node_input(); in SetupGraph() local181 node_input.set_node_id(node_info->node_id()); in SetupGraph()182 node_input.set_output_port(port); in SetupGraph()215 const GraphTransferNodeInput& node_input = input_params.node_input(i); in SetupGraph() local216 node_ids[i] = node_input.node_id() + NODE_ID_OFFSET; in SetupGraph()217 ports[i] = node_input.output_port(); in SetupGraph()
247 for (const GraphTransferNodeInput& 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()
195 for (auto& node_input : *function_node.mutable_input()) { in GetUniqueSignature()196 const auto& input = ParseNodeConnection(node_input); in GetUniqueSignature()199 node_input = *new_name + ParseOutputNode(node_input); in GetUniqueSignature()216 for (auto& node_input : *function_node.mutable_input()) { in FuseFunctionNodes()217 auto parsed_name = ParseNodeConnection(node_input); in FuseFunctionNodes()224 node_input = in FuseFunctionNodes()
114 for (input_index, node_input) in enumerate(node.input):115 if node_input == "b":117 if node_input == "y":
360 for node_input in node.input:361 if "^" in node_input:362 control_input_names.add(node_input.replace("^", ""))
33 repeated GraphTransferNodeInput node_input = 2; field
774 const string& node_input = node_def->input(i); in AddControlDependencies() local775 if (node_input[0] != '^') { in AddControlDependencies()780 if (node_input == input) { in AddControlDependencies()