Home
last modified time | relevance | path

Searched full:bmm (Results 1 – 25 of 355) sorted by relevance

12345678910>>...15

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/
H A D_sfdp_pattern_16.py44 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
61 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
66 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
77 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
81 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
86 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
108 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
123 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
136 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
152 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_18.py44 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
63 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
68 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
81 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
85 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
90 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
113 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
130 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
147 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
165 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_13.py36 bmm_default = CallFunction(aten.bmm.default, KeywordArg('query'), permute_default, _users=2)
44 bmm_default_1 = CallFunction(aten.bmm.default, mul_Tensor_1, KeywordArg('value'))
47 bmm_default_2 = CallFunction(aten.bmm.default, KeywordArg('tangents_1'), permute_default_1)
55 bmm_default_3 = CallFunction(aten.bmm.default, fma_default, permute_default_2)
57 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, fma_default)
60 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, KeywordArg('tangents_1'))
70 bmm_default = CallFunction(aten.bmm.default, KeywordArg('query'), permute_default, _users=2)
76 _sfdp_pattern_13_inference = CallFunction(aten.bmm.default, div_Tensor, KeywordArg('value'), _users…
82 bmm_default = CallFunction(aten.bmm.default, KeywordArg('query'), permute_default)
92 bmm_default_1 = CallFunction(aten.bmm.default, mul_Tensor_1, KeywordArg('value'))
[all …]
H A D_sfdp_pattern_1.py38 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
50 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
55 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
63 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
66 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
70 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
85 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
97 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
106 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
120 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_5.py38 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
51 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
56 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
64 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
67 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
71 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
86 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
99 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
108 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
123 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_2.py38 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
50 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
55 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
63 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
66 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
70 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
85 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
97 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
106 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
120 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_6.py40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
55 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
60 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
71 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
74 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
78 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
94 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
107 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
118 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
135 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_3.py40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
54 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
59 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
70 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
73 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
77 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
93 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
105 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
116 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
132 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_4.py40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
54 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
59 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
70 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
73 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
77 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
93 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
105 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
116 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
132 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_11.py42 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
56 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
61 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
69 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
73 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
78 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
98 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
112 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
125 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
141 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_10.py43 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
57 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
62 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
71 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
76 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
81 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
101 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
115 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
129 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
144 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_8.py42 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
57 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
62 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
72 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
76 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
81 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
100 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
115 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
128 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
144 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_14.py42 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
57 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
62 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
70 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
74 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
79 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
100 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
115 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
128 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
145 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_7.py44 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
61 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
66 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
79 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
83 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
88 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
108 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
123 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
138 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
156 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_19.py40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
58 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
63 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_1)
76 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_2)
79 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_3, view_default_8)
83 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_5, view_default_6)
100 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
116 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
127 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
146 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_9.py45 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
61 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
66 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
78 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
83 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
88 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
109 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
123 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
139 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
156 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_12.py44 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
60 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
65 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
76 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
80 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
85 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
106 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
120 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
135 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
153 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_15.py45 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
60 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
66 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
75 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
79 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
84 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
109 bmm_default = CallFunction(aten.bmm.default, view_default_1, view_default_2)
124 bmm_default_1 = CallFunction(aten.bmm.default, view_default_4, view_default_5)
140 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
157 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
H A D_sfdp_pattern_17.py47 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
64 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
70 bmm_default_2 = CallFunction(aten.bmm.default, view_default_6, permute_default_4)
82 bmm_default_3 = CallFunction(aten.bmm.default, view_default_8, permute_default_5)
86 bmm_default_4 = CallFunction(aten.bmm.default, permute_default_7, view_default_8)
91 bmm_default_5 = CallFunction(aten.bmm.default, permute_default_10, view_default_6)
117 bmm_default = CallFunction(aten.bmm.default, view_default_1, view_default_2)
132 bmm_default_1 = CallFunction(aten.bmm.default, view_default_4, view_default_5)
150 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
169 bmm_default_1 = CallFunction(aten.bmm.default, view_default_3, view_default_4)
[all …]
/aosp_15_r20/external/executorch/backends/qualcomm/_passes/
H A Dconvert_bmm_to_matmul.py18 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.
/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_bmm.py23 class BMM(torch.nn.Module): class in TestBMM
33 return torch.bmm(x, y)
50 return torch.bmm(x, x)
107 .check_count({"torch.ops.aten.bmm.default": 1})
115 @parameterized.expand(BMM.test_parameters)
118 self._test_bmm_tosa_MI_pipeline(self.BMM(), test_data)
135 @parameterized.expand(BMM.test_parameters)
138 self._test_bmm_tosa_BI_pipeline(self.BMM(), test_data)
145 @parameterized.expand(BMM.test_parameters)
148 self._test_bmm_tosa_BI_pipeline(self.BMM(), test_data)
/aosp_15_r20/frameworks/base/services/backup/java/com/android/server/backup/utils/
H A DBackupManagerMonitorDumpsysUtils.java45 * Util class to parse a BMM event and write it to a text file, to be the printed in
53 // Name of the subdirectory where the text file containing the BMM events will be stored.
57 // Retention period of 60 days (in millisec) for the BMM Events.
58 // After tha time has passed the text file containing the BMM events will be emptied
60 // Size limit for the text file containing the BMM events
152 Slog.e(TAG, "IO Exception when writing BMM events to file: " + e); in parseBackupManagerMonitorRestoreEventForDumpsys()
249 * Get the path of the text files which stores the BMM events
406 * Store the timestamp when the device was set up (date when the first BMM event is parsed)
502 * Delete the BMM Events file after the retention period has passed.
506 * expired BMM events from being printed to the dumpsys
[all …]
/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dbmm.py32 name="bmm",
73 idx_q = tl.program_id(1) # batch dimension for BMM
92 idx_q = tl.program_id(1) # batch dimension for BMM
102 aten_bmm = ExternKernelChoice(torch.bmm, "at::bmm_out")
106 @L.register_lowering(aten.bmm)
127 # Make the input of bmm contiguous
129 # because bmm cpu implementation would do contiguous() if not.
130 # This is to avoid additional copies in bmm.
167 return autotune_select_algorithm("bmm", choices, [mat1, mat2], layout)
/aosp_15_r20/external/executorch/exir/tests/
H A Dmodels.py203 # transposed. If yes, transpose it back before feeding to torch.bmm
208 return torch.bmm(x, y.transpose(-1, -2))
210 return torch.bmm(x, y)
331 self.bmm = BatchMatrixMultiplication(transposed=False)
369 # bmm((B*H, L, D/H), (B*H, D/H, S)) -> (B*H, L, S).
370 # this is equiv. to `qk = torch.bmm(q, k.transpose(-1, -2))`
378 # bmm((B*H, L, S), (B*H, S, D/H)) -> (B*H, L, D/H).
379 # this is equiv. to `attention = torch.bmm(softmax_qk, v)`
380 attention = self.bmm(softmax_qk, v)
449 # bmm((B*H, L, D/H), (B*H, D/H, S)) -> (B*H, L, S).
[all …]
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dgroup_batch_fusion.py162 Fuse ops in a batch way, e.g, fuse mm/addmm of same input shapes with bmm.
254 aten.bmm,
257 fused_bmm.meta["val"] = aten.bmm(
605 # the mm -> bmm transform adds an unbind() op,
686 bmm = graph.call_function(
687 torch.bmm,
690 bmm.meta["example_value"] = torch.bmm(
694 bmm_meta = bmm.meta["example_value"]
706 bmm = graph.call_function(
713 bmm.meta["example_value"] = torch.baddbmm(
[all …]

12345678910>>...15