Searched refs:input_match (Results 1 – 4 of 4) sorted by relevance
45 bool input_match = true; in TRTEngineOpShapeInference() local53 input_match = false; in TRTEngineOpShapeInference()70 if (input_match) c->set_output(i, handle); in TRTEngineOpShapeInference()
132 for input_match in args:133 self.input_op_matches.append(_make_graph_match(input_match))141 for input_match in args:142 self.control_input_op_matches.append(_make_graph_match(input_match))
34 for (const NodeMatch& input_match : match.inputs) { in RecordMatchedNodes() local35 RecordMatchedNodes(input_match, matched_nodes); in RecordMatchedNodes()55 for (const NodeMatch& input_match : current_match.inputs) { in MatchedNodesAsArray() local56 next_matches.push_back(input_match); in MatchedNodesAsArray()352 NodeMatch* input_match = &(match->inputs.back()); in DoesOpTypeMatch() local354 input_match)) { in DoesOpTypeMatch()
1270 bool input_match = false; in FuseRemoteGraphByPlacedArguments() local1284 input_match = true; in FuseRemoteGraphByPlacedArguments()1289 input_match = true; in FuseRemoteGraphByPlacedArguments()1293 if (!input_match) { in FuseRemoteGraphByPlacedArguments()