| D | convert_bmm_to_matmul.py | 18 Replace bmm to matmul, because bmm is eqaul to matmul in QNN.
19 Handle missing quantization tag for bmm op.
25 bmm = exir_ops.edge.aten.bmm.default variable in ConvertBmmToMatmul
28 {expand_copy: 2, view_copy: 3, bmm: 1},
29 {expand_copy: 2, view_copy: 3, bmm: 1, clone: 1},
30 {bmm: 1},
49 graph, [operator.matmul, torch.matmul, torch.bmm]
58 bmm_node = [n for n in src_partition.nodes if n.target == self.bmm][0]
63 # replace bmm to matmul, because bmm is eqaul to matmul in qnn.
|