Home
last modified time | relevance | path

Searched refs:get_source_partitions (Results 1 – 22 of 22) sorted by relevance

/external/executorch/backends/qualcomm/_passes/
Dannotate_decomposed.py9 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
25 partitions = get_source_partitions(graph_module.graph, [torch.unbind, "unbind"])
35 partitions = get_source_partitions(graph_module.graph, [torch.stack])
Dconvert_prelu.py9 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
23 partitions = get_source_partitions(graph, [torch.nn.PReLU])
Dreduce_dynamic_range.py12 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
40 src_partitions = get_source_partitions(
Ddecompose_silu.py10 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
25 partitions = get_source_partitions(graph, [torch.nn.functional.silu])
Dconvert_interpolate_with_upsample2d.py9 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
24 partitions = get_source_partitions(graph, [torch.nn.functional.interpolate])
Drecompose_rms_norm.py9 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
37 partitions = get_source_partitions(graph, [torch.nn.RMSNorm])
Dconvert_bmm_to_matmul.py13 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
48 partitions = get_source_partitions(
Dannotate_and_quant_scalar.py15 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
88 src_partitions = get_source_partitions(
Dconvert_to_linear.py20 get_source_partitions,
193 partitions = get_source_partitions(graph_module.graph, [torch.nn.Linear])
/external/pytorch/test/fx/
Dtest_source_matcher_utils.py16 get_source_partitions,
46 module_partitions = get_source_partitions(
104 module_partitions = get_source_partitions(
181 module_partitions = get_source_partitions(
207 module_partitions = get_source_partitions(
261 module_partitions = get_source_partitions(gm.graph, ["linear", "relu"])
323 module_partitions = get_source_partitions(
408 module_partitions = get_source_partitions(gm.graph, ["conv2d"])
440 module_partitions = get_source_partitions(gm.graph, ["linear", "relu"])
/external/executorch/backends/arm/quantizer/quantization_annotation/
Dupsample_nearest2d_annotator.py19 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
39 module_partitions = get_source_partitions(
Dmm_annotator.py18 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
27 mm_partitions = get_source_partitions(
Dadaptive_ang_pool2d_annotator.py22 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
32 module_partitions = get_source_partitions(
Dmax_pool2d_annotator.py21 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
30 module_partitions = get_source_partitions(
/external/executorch/backends/arm/test/quantizer/
Dtest_generic_annotater.py13 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
38 partitions = get_source_partitions(quant_model.graph, [model.op])
/external/pytorch/torch/ao/quantization/pt2e/
Dgraph_utils.py10 get_source_partitions,
104 partitions = get_source_partitions(gm.graph, types_to_match, filter_fn)
/external/executorch/backends/xnnpack/_passes/
Dconvert_to_linear.py21 get_source_partitions,
156 src_partition_dict = get_source_partitions(
/external/pytorch/torch/fx/passes/utils/
Dsource_matcher_utils.py51 def get_source_partitions( function
/external/executorch/backends/xnnpack/partition/config/
Dgemm_configs.py38 get_source_partitions,
352 self.src_partitions = get_source_partitions(ep.graph, self.linear_modules)
/external/pytorch/torch/ao/quantization/quantizer/
Dxnnpack_quantizer_utils.py33 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
590 gru_partitions = get_source_partitions(gm.graph, [torch.nn.GRU], filter_fn)
643 module_partitions = get_source_partitions(
961 cat_partitions = get_source_partitions(gm.graph, [torch.cat], filter_fn)
Dx86_inductor_quantizer.py54 get_source_partitions,
1253 conv_partitions = get_source_partitions(
1473 linear_partitions = get_source_partitions(
/external/executorch/docs/source/
Dcompiler-custom-compiler-passes.md472 `get_source_partitions(graph: torch.fx.Graph, wanted_sources: List[Any]) -> Dict[Any, SourcePartiti…
539 module_partitions = get_source_partitions(edge_graph, [torch.nn.Linear, torch.nn.ReLU])