(passes_job)
| 306 | |
| 307 | |
| 308 | def get_decomp_table(passes_job) -> Dict[torch._ops.OperatorBase, Callable]: |
| 309 | source_decompositions = core_aten_decompositions() |
| 310 | # The below super ops are supported by QNN |
| 311 | skip_decompositions = get_skip_decomp_table() |
| 312 | |
| 313 | # If we want to annotate the decomposed ops, then we should decompose the operation. |
| 314 | if passes_job: |
| 315 | skip_decompositions = [ |
| 316 | skip_decomp_op |
| 317 | for skip_decomp_op in skip_decompositions |
| 318 | if skip_decomp_op |
| 319 | not in AnnotateStack.decomp_ops + AnnotateUnbind.decomp_ops |
| 320 | ] |
| 321 | remove_decompositions(source_decompositions, skip_decompositions) |
| 322 | |
| 323 | return source_decompositions |
| 324 | |
| 325 | |
| 326 | def to_edge_transform_and_lower_to_qnn( |
no test coverage detected