Searched refs:generate_pattern_with_output_quant (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | quantization.py | 262 def generate_pattern_with_output_quant(computation_call, with_dtype_convert=False): function
722 UnaryAttr("none", [], ""): generate_pattern_with_output_quant(
725 UnaryAttr("relu", [], ""): generate_pattern_with_output_quant(
730 UnaryAttr("hardtanh", [], ""): generate_pattern_with_output_quant(
739 UnaryAttr("hardswish", [], ""): generate_pattern_with_output_quant(
748 UnaryAttr("swish", [], ""): generate_pattern_with_output_quant(
819 UnaryAttr("none", [], ""): generate_pattern_with_output_quant(
822 UnaryAttr("relu", [], ""): generate_pattern_with_output_quant(
825 UnaryAttr("gelu", [], "none"): generate_pattern_with_output_quant(
836 UnaryAttr("gelu", [], "tanh"): generate_pattern_with_output_quant(
[all …]
|