Searched refs:NodeNameFromInput (Results 1 – 8 of 8) sorted by relevance
122 required_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()123 input_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()134 current_inputs.push_back(NodeNameFromInput(output_name)); in StripUnusedNodes()150 string input_node_name = NodeNameFromInput(input_name); in StripUnusedNodes()
78 string input_node_name = NodeNameFromInput(input); in MapNodesToOutputs()101 string NodeNameFromInput(const string& input_name) { in NodeNameFromInput() function219 const string& input_node_name = NodeNameFromInput(input_name); in SortByExecutionOrder()350 const string& input_node_name = NodeNameFromInput(non_control_inputs[i]); in DoesOpTypeMatch()410 string input_node_name = NodeNameFromInput(input_name); in ReplaceMatchingOpTypes()484 canonical_inputs_to_rename[NodeNameFromInput(input_to_rename.first)] in RenameNodeInputs()497 canonical_inputs_to_rename.count(NodeNameFromInput(new_input_name))) { in RenameNodeInputs()498 string input_node_name = NodeNameFromInput(new_input_name); in RenameNodeInputs()564 string input_node = NodeNameFromInput(input); in FindInvalidInputs()
45 required_nodes.insert(NodeNameFromInput(input)); in RemoveNodes()48 required_nodes.insert(NodeNameFromInput(output)); in RemoveNodes()
131 NodeNameFromInput(node_map.at("requantize_op")->input(3)); in TestFreezeRequantizationRanges()135 NodeNameFromInput(node_map.at("requantize_op")->input(4)); in TestFreezeRequantizationRanges()
90 NodeNameFromInput(q_weights_op->input(0)); in TestQuantizeWeights()
266 graph_outputs.insert(NodeNameFromInput(output_name)); in RemoveRedundantQuantizations()750 AddNodeInput("^" + NodeNameFromInput(input_name), &reshape_dims); in QuantizeNodes()760 AddNodeInput("^" + NodeNameFromInput(input_name), &reduction_dims); in QuantizeNodes()
59 string NodeNameFromInput(const string& input_name);
144 EXPECT_EQ("node_name", NodeNameFromInput("node_name")); in TestNodeNameFromInput()145 EXPECT_EQ("node_name", NodeNameFromInput("node_name:0")); in TestNodeNameFromInput()146 EXPECT_EQ("node_name", NodeNameFromInput("^node_name")); in TestNodeNameFromInput()147 EXPECT_EQ("node_name", NodeNameFromInput("^node_name:42")); in TestNodeNameFromInput()