Searched defs:computation_op (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | mkldnn_fusion.py | 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
382 def _is_valid_computation_binary(computation_op, binary_op, other_index=None): argument
433 def _is_valid_computation_binary_inplace(computation_op, binary_op, other_index): argument
471 computation_op, argument
510 computation_op, argument
|
| H A D | quantization.py | 308 computation_op, argument
395 computation_op, argument
453 computation_op, argument
625 computation_op, argument
1235 computation_op, argument
1374 computation_op, argument
1415 computation_op, argument
|