Home
last modified time | relevance | path

Searched refs:filter_nodes (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/torch/_inductor/fx_passes/
Dmkldnn_fusion.py15 filter_nodes,
202 computation_nodes = filter_nodes(match.nodes, computation_op)
220 computation_node = filter_nodes(match.nodes, computation_op)[0]
222 conversion_dtype_nodes = filter_nodes(
347 binary_nodes = filter_nodes(match.nodes, fn)
437 binary_nodes = filter_nodes(match.nodes, binary_op)
Dquantization.py16 from ..pattern_matcher import Arg, CallFunction, filter_nodes, KeywordArg, ListOf, Match
294 qconv_node_after_weight_prepack = filter_nodes(
381 qlinear_node_after_weight_prepack = filter_nodes(
558 compute_node = filter_nodes(match.nodes, qop)[0]
1227 get_item_node = filter_nodes(match.nodes, operator.getitem)[0]
1348 dequant_nodes = filter_nodes(
1353 quant_nodes = filter_nodes(
1957 linear_nodes = filter_nodes(match.nodes, aten.bmm.default)
Dpost_grad.py32 filter_nodes,
607 split_nodes = filter_nodes(match.nodes, aten.split_with_sizes)
608 cat_nodes = filter_nodes(match.nodes, aten.cat)
609 get_item_nodes = filter_nodes(match.nodes, operator.getitem)
921 cat_nodes = filter_nodes(match.nodes, aten.cat)
922 split_nodes = filter_nodes(match.nodes, aten.split_with_sizes)
Dfuse_attention.py12 filter_nodes,
562 add_mask_node = filter_nodes(match.nodes, aten.add.Tensor)
594 scale_factor_node = filter_nodes(match.nodes, scale_factor_op)[0]
/external/executorch/exir/tests/
Dtest_memory_planning.py18 filter_nodes,
414 filter_nodes(
/external/executorch/exir/
Dmemory_planning.py288 def filter_nodes(inputs: Iterable[Any]) -> Iterable[Node]: function
438 filter_nodes(itertools.chain([node], node.args, node.kwargs.values())),
/external/pytorch/torch/_inductor/
Dpattern_matcher.py1990 def filter_nodes(nodes: Iterable[torch.fx.Node], fn: Any) -> List[torch.fx.Node]: function