Searched refs:_unary_fusion_pattern (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | quantization.py | 112 def _unary_fusion_pattern(unary_fusion, call_fn, users, is_bf16): function
731 _unary_fusion_pattern(
740 _unary_fusion_pattern(
749 _unary_fusion_pattern(
774 _unary_fusion_pattern(
784 _unary_fusion_pattern(
794 _unary_fusion_pattern(
826 _unary_fusion_pattern(
837 _unary_fusion_pattern(
863 _unary_fusion_pattern(
[all …]
|
| H A D | mkldnn_fusion.py | 80 def _unary_fusion_pattern(unary_fusion, call_fn, users, lowp_dtype): function
567 _unary_fusion_pattern(_gelu_fusion_2, call_fn, 4, lowp_dtype)
571 _unary_fusion_pattern(_gelu_fusion_1, call_fn, 2, lowp_dtype)
575 _unary_fusion_pattern(_hardswish_fusion, call_fn, 2, lowp_dtype)
579 _unary_fusion_pattern(_hardsigmoid_fusion, call_fn, 1, lowp_dtype)
583 _unary_fusion_pattern(_silu_fusion, call_fn, 2, lowp_dtype)
618 _unary_fusion_pattern(_leaky_relu_fusion, call_fn, 3, lowp_dtype)
626 _unary_fusion_pattern(_hardtanh_fusion, call_fn, 1, lowp_dtype)
|