Searched refs:_get_aten_graph_module_for_pattern (Results 1 – 5 of 5) sorted by relevance
50 from .utils import _get_aten_graph_module_for_pattern66 match_pattern = _get_aten_graph_module_for_pattern(69 replacement_pattern = _get_aten_graph_module_for_pattern(73 match_pattern = _get_aten_graph_module_for_pattern(76 replacement_pattern = _get_aten_graph_module_for_pattern(105 from .utils import _get_aten_graph_module_for_pattern143 bn_train_aten = _get_aten_graph_module_for_pattern(148 bn_eval_aten = _get_aten_graph_module_for_pattern(
23 _get_aten_graph_module_for_pattern,670 match_pattern = _get_aten_graph_module_for_pattern(681 replacement_pattern_with_conv_bias = _get_aten_graph_module_for_pattern(698 replacement_pattern_no_conv_bias = _get_aten_graph_module_for_pattern(911 match_pattern = _get_aten_graph_module_for_pattern(917 replacement_pattern = _get_aten_graph_module_for_pattern(
362 def _get_aten_graph_module_for_pattern( function
11 _get_aten_graph_module_for_pattern,816 …pattern = _get_aten_graph_module_for_pattern(pattern, example_inputs) # type: ignore[arg-type, as…818 …replacement = _get_aten_graph_module_for_pattern(replacement, example_inputs) # type: ignore[arg-…
15 _get_aten_graph_module_for_pattern,526 …pattern = _get_aten_graph_module_for_pattern(pattern, example_inputs, is_cuda) # type: ignore[has…