Searched refs:unary_algorithm (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | mkldnn_ir.py | 361 unary_algorithm: Optional[str],
378 unary_algorithm,
457 unary_algorithm: Optional[str],
474 unary_algorithm,
663 unary_algorithm,
694 unary_algorithm,
916 unary_algorithm,
954 unary_algorithm,
1024 unary_algorithm, argument
1074 unary_algorithm,
[all …]
|
| H A D | mkldnn_lowerings.py | 111 unary_algorithm, argument
127 unary_algorithm,
145 unary_algorithm, argument
161 unary_algorithm,
|
| H A D | ir.py | 6118 unary_algorithm: Optional[str],
6135 unary_algorithm,
6213 unary_algorithm: Optional[str],
6230 unary_algorithm,
6699 unary_algorithm,
6719 unary_algorithm,
6751 unary_algorithm, argument
6784 unary_algorithm,
6881 unary_algorithm,
6905 unary_algorithm,
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv.cpp | 1410 std::optional<c10::string_view> unary_algorithm) { in _quantized_convolution_onednn() argument
1615 unary_algorithm.has_value() ? unary_algorithm.value() : "" in _quantized_convolution_onednn()
1940 std::optional<c10::string_view> unary_algorithm) { in run_pointwise_binary() argument
1964 unary_attr, unary_scalars, unary_algorithm in run_pointwise_binary()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 348 std::optional<c10::string_view> unary_algorithm) { in mkldnn_convolution_pointwise_binary() argument
512 std::optional<c10::string_view> unary_algorithm) { in mkldnn_convolution_pointwise_binary_() argument
|