| /external/pytorch/torch/_inductor/fx_passes/serialized_patterns/ |
| D | _sfdp_pattern_16.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
36 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
37 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
38 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
41 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
[all …]
|
| D | _sfdp_pattern_18.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
36 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
37 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
38 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
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())
41 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
[all …]
|
| D | _sfdp_pattern_9.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
36 div_Tensor = CallFunction(aten.div.Tensor, permute_default, Ignored())
37 expand_default = CallFunction(aten.expand.default, div_Tensor, Ignored())
38 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
39 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
40 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
41 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
[all …]
|
| D | _sfdp_pattern_7.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
36 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
37 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
38 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
41 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
[all …]
|
| D | _sfdp_pattern_11.py | 16 CallFunction,
33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
34 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
35 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
36 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
37 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
39 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
40 clone_default_1 = CallFunction(aten.clone.default, expand_default_1, memory_format=torch.contiguous…
41 view_default_1 = CallFunction(aten.view.default, clone_default_1, Ignored(), _users=2)
[all …]
|
| D | _sfdp_pattern_12.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
36 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
37 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
38 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
39 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
40 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
41 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
[all …]
|
| D | _sfdp_pattern_15.py | 16 CallFunction,
33 eq_Scalar = CallFunction(aten.eq.Scalar, KeywordArg('attn_mask'), Ignored())
34 expand_default = CallFunction(aten.expand.default, eq_Scalar, Ignored(), _users=2)
35 full_default = CallFunction(aten.full.default, [], Ignored(), dtype=Ignored(), device=Ignored(), pi…
36 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
37 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
38 clone_default = CallFunction(aten.clone.default, expand_default_1, memory_format=torch.contiguous_f…
39 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
40 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
41 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
[all …]
|
| D | _sfdp_pattern_17.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 eq_Scalar = CallFunction(aten.eq.Scalar, KeywordArg('attn_mask'), Ignored())
36 expand_default = CallFunction(aten.expand.default, eq_Scalar, Ignored(), _users=2)
37 full_default = CallFunction(aten.full.default, [], Ignored(), dtype=Ignored(), device=Ignored(), pi…
38 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
39 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
40 clone_default = CallFunction(aten.clone.default, expand_default_1, memory_format=torch.contiguous_f…
41 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
[all …]
|
| D | _sfdp_pattern_3.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
36 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
37 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
39 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
41 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
[all …]
|
| D | _sfdp_pattern_4.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
36 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
37 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
39 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
41 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
[all …]
|
| D | _sfdp_pattern_14.py | 16 CallFunction,
33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
34 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
35 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
36 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
37 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
39 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
40 clone_default_1 = CallFunction(aten.clone.default, expand_default_1, memory_format=torch.contiguous…
41 view_default_1 = CallFunction(aten.view.default, clone_default_1, Ignored(), _users=2)
[all …]
|
| D | _sfdp_pattern_10.py | 16 CallFunction,
33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
34 div_Tensor = CallFunction(aten.div.Tensor, permute_default, Ignored())
35 expand_default = CallFunction(aten.expand.default, div_Tensor, Ignored())
36 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
37 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
38 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
39 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
40 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
41 clone_default_1 = CallFunction(aten.clone.default, expand_default_1, memory_format=torch.contiguous…
[all …]
|
| D | _sfdp_pattern_8.py | 16 CallFunction,
33 permute_default = CallFunction(aten.permute.default, KeywordArg('query'), Ignored())
34 expand_default = CallFunction(aten.expand.default, permute_default, Ignored())
35 clone_default = CallFunction(aten.clone.default, expand_default, memory_format=torch.contiguous_for…
36 view_default = CallFunction(aten.view.default, clone_default, Ignored(), _users=2)
37 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 permute_default_2 = CallFunction(aten.permute.default, permute_default_1, Ignored())
39 expand_default_1 = CallFunction(aten.expand.default, permute_default_2, Ignored())
40 clone_default_1 = CallFunction(aten.clone.default, expand_default_1, memory_format=torch.contiguous…
41 view_default_1 = CallFunction(aten.view.default, clone_default_1, Ignored(), _users=2)
[all …]
|
| D | _sfdp_pattern_5.py | 16 CallFunction,
33 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
34 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
36 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
37 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
38 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
39 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
40 div_Tensor = CallFunction(aten.div.Tensor, view_default_2, Ignored())
41 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2)
[all …]
|
| D | _sfdp_pattern_6.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
36 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
37 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
39 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
41 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
[all …]
|
| D | _sfdp_pattern_19.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
36 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
37 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
38 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
39 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
40 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
41 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
[all …]
|
| D | _sfdp_pattern_1.py | 16 CallFunction,
33 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
34 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
36 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
37 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
38 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
39 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
40 div_Tensor = CallFunction(aten.div.Tensor, view_default_2, KeywordArg('inv_scale'), _users=2)
41 amax_default = CallFunction(aten.amax.default, div_Tensor, Ignored(), True)
[all …]
|
| D | _sfdp_pattern_2.py | 16 CallFunction,
33 expand_default = CallFunction(aten.expand.default, KeywordArg('query'), Ignored())
34 view_default = CallFunction(aten.view.default, expand_default, Ignored(), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored())
36 expand_default_1 = CallFunction(aten.expand.default, permute_default, Ignored())
37 view_default_1 = CallFunction(aten.view.default, expand_default_1, Ignored(), _users=2)
38 bmm_default = CallFunction(aten.bmm.default, view_default, view_default_1)
39 view_default_2 = CallFunction(aten.view.default, bmm_default, Ignored())
40 mul_Tensor = CallFunction(aten.mul.Tensor, view_default_2, KeywordArg('scale_factor'), _users=2)
41 amax_default = CallFunction(aten.amax.default, mul_Tensor, Ignored(), True)
[all …]
|
| D | _sfdp_pattern_13.py | 16 CallFunction,
33 rand_default = CallFunction(aten.rand.default, Ignored(), dtype=Ignored(), device=Ignored(), pin_me…
34 gt_Scalar = CallFunction(aten.gt.Scalar, rand_default, KeywordArg('dropout_p'), _users=2)
35 permute_default = CallFunction(aten.permute.default, KeywordArg('key'), Ignored(), _users=2)
36 bmm_default = CallFunction(aten.bmm.default, KeywordArg('query'), permute_default, _users=2)
37 amax_default = CallFunction(aten.amax.default, bmm_default, Ignored(), True)
38 sub_Tensor = CallFunction(aten.sub.Tensor, bmm_default, amax_default)
39 exp_default = CallFunction(aten.exp.default, sub_Tensor, _users=2)
40 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
41 div_Tensor = CallFunction(aten.div.Tensor, exp_default, sum_dim_IntList, _users=3)
[all …]
|
| D | addmm_pattern.py | 16 CallFunction,
33 addmm_default = CallFunction(aten.addmm.default, KeywordArg('input'), KeywordArg('mat1'), KeywordAr…
34 mul_Scalar = CallFunction(aten.mul.Scalar, KeywordArg('tangents_1'), KeywordArg('beta'))
35 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, mul_Scalar, Ignored(), True)
36 view_default = CallFunction(aten.view.default, sum_dim_IntList, Ignored())
37 permute_default = CallFunction(aten.permute.default, KeywordArg('mat2'), Ignored())
38 mm_default = CallFunction(aten.mm.default, KeywordArg('tangents_1'), permute_default)
39 mul_Scalar_1 = CallFunction(aten.mul.Scalar, mm_default, KeywordArg('alpha'))
40 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('mat1'), Ignored())
41 mm_default_1 = CallFunction(aten.mm.default, permute_default_1, KeywordArg('tangents_1'))
[all …]
|
| D | mm_pattern.py | 16 CallFunction,
33 mm_default = CallFunction(aten.mm.default, KeywordArg('mat1'), KeywordArg('mat2'))
34 permute_default = CallFunction(aten.permute.default, KeywordArg('mat2'), Ignored())
35 mm_default_1 = CallFunction(aten.mm.default, KeywordArg('tangents_1'), permute_default)
36 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('mat1'), Ignored())
37 mm_default_2 = CallFunction(aten.mm.default, permute_default_1, KeywordArg('tangents_1'))
44 mm_pattern_inference = CallFunction(aten.mm.default, KeywordArg('mat1'), KeywordArg('mat2'), _users…
|
| D | bmm_pattern.py | 16 CallFunction,
33 bmm_default = CallFunction(aten.bmm.default, KeywordArg('mat1'), KeywordArg('mat2'))
34 permute_default = CallFunction(aten.permute.default, KeywordArg('mat2'), Ignored())
35 bmm_default_1 = CallFunction(aten.bmm.default, KeywordArg('tangents_1'), permute_default)
36 permute_default_1 = CallFunction(aten.permute.default, KeywordArg('mat1'), Ignored())
37 bmm_default_2 = CallFunction(aten.bmm.default, permute_default_1, KeywordArg('tangents_1'))
44 bmm_pattern_inference = CallFunction(aten.bmm.default, KeywordArg('mat1'), KeywordArg('mat2'), _use…
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | mkldnn_fusion.py | 14 CallFunction,
40 return CallFunction(
45 return CallFunction(
50 return CallFunction(
57 return CallFunction(
65 return CallFunction(
73 return CallFunction(
94 return CallFunction(
96 CallFunction(aten.mul, computation_call, 0.5),
97 CallFunction(
[all …]
|
| D | post_grad.py | 30 CallFunction,
260 CallFunction(
262 CallFunction(
309 CallFunction(
311 CallFunction(aten.mm, KeywordArg("mat1"), KeywordArg("mat2")),
312 CallFunction(aten.mm, KeywordArg("mat3"), KeywordArg("mat4")),
361 CallFunction(
364 CallFunction(
366 CallFunction(
368 CallFunction(
[all …]
|
| D | quantization.py | 16 from ..pattern_matcher import Arg, CallFunction, filter_nodes, KeywordArg, ListOf, Match
74 return CallFunction(
86 return CallFunction(
100 t_pattern = CallFunction(
121 dequantize_per_tensor_activation_pattern = CallFunction(
135 dequantize_per_channel_weight_pattern = CallFunction(
153 dequantize_per_channel_clone_weight_pattern = CallFunction(
159 dequantize_per_channel_to_bf16_clone_weight_pattern = CallFunction(
167 return CallFunction(
196 return CallFunction(
[all …]
|