| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/ |
| H A D | _sfdp_pattern_16.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
57 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
65 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
76 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
79 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
80 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
83 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
84 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_18.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored(), _users=2)
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
59 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored(), _users=2)
67 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
80 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
83 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
84 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
87 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
88 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_11.py | 33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
37 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
52 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
60 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
68 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
71 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
72 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
75 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
76 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_10.py | 33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
38 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
39 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
53 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
61 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
70 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
74 permute_default_6 = CallFunction(aten.permute.default, div_Tensor_2, Ignored())
75 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
78 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
79 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_8.py | 33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
37 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
53 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
61 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
71 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
74 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
75 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
78 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
79 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_14.py | 33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
37 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
53 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
61 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
69 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
72 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
73 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
76 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
77 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_7.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
57 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
65 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
78 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
81 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
82 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
85 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
86 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_9.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
40 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
41 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
57 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
65 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
77 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
81 permute_default_6 = CallFunction(aten.permute.default, div_Tensor_2, Ignored())
82 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
85 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
86 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_12.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
56 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
64 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
75 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
78 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
79 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
82 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
83 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_15.py | 36 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
40 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
41 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
56 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
65 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
74 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
77 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
78 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
81 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
82 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_17.py | 38 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
42 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
43 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
60 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
69 permute_default_4 = CallFunction(aten.permute.default, view_default_4, Ignored())
81 permute_default_5 = CallFunction(aten.permute.default, view_default_1, Ignored())
84 permute_default_6 = CallFunction(aten.permute.default, view_default_9, Ignored())
85 permute_default_7 = CallFunction(aten.permute.default, view_default, Ignored())
88 permute_default_8 = CallFunction(aten.permute.default, view_default_10, Ignored())
89 permute_default_9 = CallFunction(aten.permute.default, permute_default_8, Ignored())
[all …]
|
| H A D | _sfdp_pattern_13.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored(), _users=2)
46 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
54 permute_default_2 = CallFunction(aten.permute.default, permute_default, Ignored())
56 permute_default_3 = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
58 permute_default_4 = CallFunction(aten.permute.default, bmm_default_4, Ignored())
59 permute_default_5 = CallFunction(aten.permute.default, mul_Tensor_1, Ignored())
69 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
81 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored(), _users=2)
95 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('value'), Ignored())
105 permute_default_2 = CallFunction(aten.permute.default, permute_default, Ignored())
[all …]
|
| H A D | _sfdp_pattern_1.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
54 permute_default_1 = CallFunction(aten.permute.default, view_default_4, Ignored())
62 permute_default_2 = CallFunction(aten.permute.default, view_default_1, Ignored())
65 permute_default_3 = CallFunction(aten.permute.default, view_default, Ignored())
68 permute_default_4 = CallFunction(aten.permute.default, view_default_10, Ignored())
69 permute_default_5 = CallFunction(aten.permute.default, view_default_3, Ignored())
82 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
103 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
125 permute_default_1 = CallFunction(aten.permute.default, view_default_4, Ignored())
135 permute_default_2 = CallFunction(aten.permute.default, view_default_1, Ignored())
[all …]
|
| H A D | _sfdp_pattern_5.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
55 permute_default_1 = CallFunction(aten.permute.default, view_default_4, Ignored())
63 permute_default_2 = CallFunction(aten.permute.default, view_default_1, Ignored())
66 permute_default_3 = CallFunction(aten.permute.default, view_default, Ignored())
69 permute_default_4 = CallFunction(aten.permute.default, view_default_10, Ignored())
70 permute_default_5 = CallFunction(aten.permute.default, view_default_3, Ignored())
83 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
105 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
128 permute_default_1 = CallFunction(aten.permute.default, view_default_4, Ignored())
138 permute_default_2 = CallFunction(aten.permute.default, view_default_1, Ignored())
[all …]
|
| H A D | _sfdp_pattern_2.py | 35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
54 permute_default_1 = CallFunction(aten.permute.default, view_default_4, Ignored())
62 permute_default_2 = CallFunction(aten.permute.default, view_default_1, Ignored())
65 permute_default_3 = CallFunction(aten.permute.default, view_default, Ignored())
68 permute_default_4 = CallFunction(aten.permute.default, view_default_10, Ignored())
69 permute_default_5 = CallFunction(aten.permute.default, view_default_3, Ignored())
82 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
103 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
125 permute_default_1 = CallFunction(aten.permute.default, view_default_4, Ignored())
135 permute_default_2 = CallFunction(aten.permute.default, view_default_1, Ignored())
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | fuse_attention.py | 163 q = query.permute(0, 2, 1, 3)
164 k = key.permute(0, 2, 1, 3)
165 v = value.permute(0, 2, 1, 3)
181 q = query.permute(0, 2, 1, 3)
182 k = key.permute(0, 2, 1, 3)
183 v = value.permute(0, 2, 1, 3)
196 q = query.permute(0, 2, 1, 3)
197 k = key.permute(0, 2, 1, 3)
198 v = value.permute(0, 2, 1, 3)
208 q = query.permute(0, 2, 1, 3)
[all …]
|
| /aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
| H A D | PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp | 31 // Insert Permute which swaps batches and channels dimensions in CreateTestNetworkImpl()
32 …auto permute = network->AddPermuteLayer(PermuteDescriptor(PermutationVector{ 3, 1, 2, 0 }), "permu… in CreateTestNetworkImpl() local
34 permute->GetOutputSlot(0).SetTensorInfo(permuteInfo); in CreateTestNetworkImpl()
35 input->GetOutputSlot(0).Connect(permute->GetInputSlot(0)); in CreateTestNetworkImpl()
44 permute->GetOutputSlot(0).Connect(batchToSpace->GetInputSlot(0)); in CreateTestNetworkImpl()
62 // Insert Permute which swaps batches and channels dimensions in CreateTransposeTestNetworkImpl()
63 …auto permute = network->AddTransposeLayer(TransposeDescriptor(PermutationVector{ 3, 1, 2, 0 }), "p… in CreateTransposeTestNetworkImpl() local
65 permute->GetOutputSlot(0).SetTensorInfo(permuteInfo); in CreateTransposeTestNetworkImpl()
66 input->GetOutputSlot(0).Connect(permute->GetInputSlot(0)); in CreateTransposeTestNetworkImpl()
75 permute->GetOutputSlot(0).Connect(batchToSpace->GetInputSlot(0)); in CreateTransposeTestNetworkImpl()
[all …]
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/tests/ |
| H A D | test_reorder_ops_passes.py | 138 x = x.permute([1, 0, 3, 2])
149 # permute node.
157 # Assert that permute node is the successor of quant node
174 x = x.permute([1, 0, 3, 2])
183 # Assert that the dequant node is no longer the predecessor of the permute node
191 # Assert that dequant node is the successor of permute node
256 # 4d -> permute -> 4d -> view -> 3d
263 x = x.permute([1, 0, 2])
267 x = x.permute([0, 1, 3, 2])
281 # Assert the order becomes view, view, permute, permute
[all …]
|
| /aosp_15_r20/external/executorch/backends/arm/test/ops/ |
| H A D | test_permute.py | 23 from torchvision.ops import Permute
37 """Tests Permute Operator."""
39 class Permute(torch.nn.Module): class in TestPermute
44 self.permute = Permute(dims=dims)
47 return self.permute(x)
64 .check(["torch.ops.aten.permute.default"])
86 .check_count({"torch.ops.aten.permute.default": 1})
111 .check_count({"torch.ops.aten.permute.default": 1})
125 self._test_permute_tosa_MI_pipeline(self.Permute(dims=dims), (test_data,), True)
127 self.Permute(dims=dims), (test_data,), False
[all …]
|
| H A D | test_select.py | 54 # For 4D tensors, do not permute to NHWC
55 permute = False if len(test_data[0].shape) == 4 else True
61 "TOSA-0.80.0+MI", permute_memory_to_nhwc=permute
80 # For 4D tensors, do not permute to NHWC
81 permute = False if len(test_data[0].shape) == 4 else True
87 "TOSA-0.80.0+BI", permute_memory_to_nhwc=permute
131 # For 4D tensors, do not permute to NHWC
132 permute = False if len(test_data[0].shape) == 4 else True
134 common.get_u55_compile_spec(permute_memory_to_nhwc=permute),
143 # For 4D tensors, do not permute to NHWC
[all …]
|
| /aosp_15_r20/external/oboe/samples/RhythmGame/third_party/glm/gtc/ |
| H A D | noise.inl | 35 tvec4<T, P> i = detail::permute(detail::permute(ix) + iy);
79 tvec4<T, P> ixy = detail::permute(detail::permute(ix) + iy);
80 tvec4<T, P> ixy0 = detail::permute(ixy + iz0);
81 tvec4<T, P> ixy1 = detail::permute(ixy + iz1);
150 tvec4<T, P> ixy = permute(permute(ix) + iy);
151 tvec4<T, P> ixy0 = permute(ixy + iz0);
152 tvec4<T, P> ixy1 = permute(ixy + iz1);
225 tvec4<T, P> ixy = detail::permute(detail::permute(ix) + iy);
226 tvec4<T, P> ixy0 = detail::permute(ixy + iz0);
227 tvec4<T, P> ixy1 = detail::permute(ixy + iz1);
[all …]
|
| /aosp_15_r20/hardware/google/gfxstream/third-party/glm/include/glm/gtc/ |
| D | noise.inl | 35 tvec4<T, P> i = detail::permute(detail::permute(ix) + iy);
79 tvec4<T, P> ixy = detail::permute(detail::permute(ix) + iy);
80 tvec4<T, P> ixy0 = detail::permute(ixy + iz0);
81 tvec4<T, P> ixy1 = detail::permute(ixy + iz1);
150 tvec4<T, P> ixy = permute(permute(ix) + iy);
151 tvec4<T, P> ixy0 = permute(ixy + iz0);
152 tvec4<T, P> ixy1 = permute(ixy + iz1);
225 tvec4<T, P> ixy = detail::permute(detail::permute(ix) + iy);
226 tvec4<T, P> ixy0 = detail::permute(ixy + iz0);
227 tvec4<T, P> ixy1 = detail::permute(ixy + iz1);
[all …]
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Permute.cpp | 24 #include "Permute.h"
38 SimpleTensor<T> permute(const SimpleTensor<T> &src, PermutationVector perm) in permute() function
40 // Permute shapes in permute()
42 permute(dst_shape, perm); in permute()
53 permute(dst_coords, perm); in permute()
61 template SimpleTensor<int8_t> permute(const SimpleTensor<int8_t> &src, PermutationVector perm);
62 template SimpleTensor<uint8_t> permute(const SimpleTensor<uint8_t> &src, PermutationVector perm);
63 template SimpleTensor<int16_t> permute(const SimpleTensor<int16_t> &src, PermutationVector perm);
64 template SimpleTensor<uint16_t> permute(const SimpleTensor<uint16_t> &src, PermutationVector perm);
65 template SimpleTensor<uint32_t> permute(const SimpleTensor<uint32_t> &src, PermutationVector perm);
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_fused_attention.py | 374 q = query.permute(0, 2, 1, 3)
375 k = key.permute(0, 2, 1, 3)
376 v = value.permute(0, 2, 1, 3)
416 q = query.permute(0, 2, 1, 3)
417 k = key.permute(0, 2, 1, 3)
418 v = value.permute(0, 2, 1, 3)
442 q = query.permute(0, 2, 1, 3)
443 k = key.permute(0, 2, 1, 3)
444 v = value.permute(0, 2, 1, 3)
484 q = query.permute(0, 2, 1, 3)
[all …]
|
| /aosp_15_r20/external/llvm/test/CodeGen/SystemZ/ |
| H A D | vec-perm-08.ll | 5 ; Test a high1/low2 permute for v16i8.
18 ; Test a low2/high1 permute for v16i8.
31 ; Test a low1/high2 permute for v16i8.
44 ; Test a high2/low1 permute for v16i8.
70 ; Test a high1/low2 permute for v8i16.
81 ; Test a low2/high1 permute for v8i16.
92 ; Test a high1/low2 permute for v4i32.
102 ; Test a low2/high1 permute for v4i32.
112 ; Test a high1/low2 permute for v2i64.
122 ; Test low2/high1 permute for v2i64.
[all …]
|