MCPcopy Create free account
hub / github.com/pytorch/executorch / get_skip_decomp_table

Function get_skip_decomp_table

backends/qualcomm/partition/utils.py:50–79  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

48
49
50def get_skip_decomp_table() -> List[torch._ops.OperatorBase]:
51 do_not_decompose = [
52 torch.ops.aten.adaptive_avg_pool2d.default,
53 torch.ops.aten.channel_shuffle.default,
54 torch.ops.aten.col2im.default,
55 torch.ops.aten.elu.default,
56 torch.ops.aten.floor_divide.default,
57 torch.ops.aten.hardsigmoid.default,
58 torch.ops.aten.hardswish.default,
59 torch.ops.aten.im2col.default,
60 torch.ops.aten.instance_norm.default,
61 torch.ops.aten.leaky_relu.default,
62 torch.ops.aten.linear.default,
63 torch.ops.aten.matmul.default,
64 torch.ops.aten.pixel_shuffle.default,
65 torch.ops.aten.pixel_unshuffle.default,
66 torch.ops.aten.prelu.default,
67 torch.ops.aten.reflection_pad1d.default,
68 torch.ops.aten.reflection_pad2d.default,
69 torch.ops.aten.rms_norm.default,
70 torch.ops.aten._safe_softmax.default,
71 torch.ops.aten.scatter.src,
72 torch.ops.aten.stack.default,
73 torch.ops.aten.upsample_bicubic2d.vec,
74 # This request is ignored because it is in a blocklist. Refer to exir/program/_program.py
75 torch.ops.aten.unbind.int,
76 torch.ops.torchao.quantize_affine.default,
77 torch.ops.torchao.dequantize_affine.default,
78 ]
79 return do_not_decompose

Callers 2

get_decomp_tableFunction · 0.85
ops_to_not_decomposeMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected