Searched defs:lowp_dtype (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | mkldnn_fusion.py | 80 def _unary_fusion_pattern(unary_fusion, call_fn, users, lowp_dtype): argument
200 def _is_single_computation_op(computation_op, lowp_dtype=None): argument
217 def _is_valid_computation_unary_fusion(computation_op, lowp_dtype=None): argument
240 pattern, unary_attr, computation_op, lowp_dtype=None argument
256 def _register_leaky_relu_fusion_lowering(pattern, computation_op, lowp_dtype=None): argument
298 def _register_hardtanh_fusion_lowering(pattern, computation_op, lowp_dtype=None): argument
564 def _unary_fusion_patterns(lowp_dtype): argument
|