Searched refs:NodeNameFromInput (Results 1 – 9 of 9) sorted by relevance
157 string input_op = node_map[NodeNameFromInput(node->input(0))]->op(); in FuseQuantizedConvolutionAndRequantize()160 NodeDef *bias_node = const_cast<NodeDef*>(node_map[NodeNameFromInput( in FuseQuantizedConvolutionAndRequantize()162 const NodeDef *min_input_node = node_map[NodeNameFromInput( in FuseQuantizedConvolutionAndRequantize()164 const NodeDef *max_input_node = node_map[NodeNameFromInput( in FuseQuantizedConvolutionAndRequantize()166 const NodeDef *min_filter_node = node_map[NodeNameFromInput( in FuseQuantizedConvolutionAndRequantize()168 const NodeDef *max_filter_node = node_map[NodeNameFromInput( in FuseQuantizedConvolutionAndRequantize()
123 required_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()124 input_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()135 current_inputs.push_back(NodeNameFromInput(output_name)); in StripUnusedNodes()151 string input_node_name = NodeNameFromInput(input_name); in StripUnusedNodes()
76 string input_node_name = NodeNameFromInput(input); in MapNodesToOutputs()99 string NodeNameFromInput(const string& input_name) { in NodeNameFromInput() function217 const string& input_node_name = NodeNameFromInput(input_name); in SortByExecutionOrder()348 const string& input_node_name = NodeNameFromInput(non_control_inputs[i]); in DoesOpTypeMatch()408 string input_node_name = NodeNameFromInput(input_name); in ReplaceMatchingOpTypes()482 canonical_inputs_to_rename[NodeNameFromInput(input_to_rename.first)] in RenameNodeInputs()495 canonical_inputs_to_rename.count(NodeNameFromInput(new_input_name))) { in RenameNodeInputs()496 string input_node_name = NodeNameFromInput(new_input_name); in RenameNodeInputs()562 string input_node = NodeNameFromInput(input); in FindInvalidInputs()
46 required_nodes.insert(NodeNameFromInput(input)); in RemoveNodes()49 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()745 AddNodeInput("^" + NodeNameFromInput(input_name), &reshape_dims); in QuantizeNodes()755 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()