Searched defs:unary_scalars (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | mkldnn_lowerings.py | 110 unary_scalars, argument
144 unary_scalars, argument
430 unary_scalars, argument
745 unary_scalars, argument
|
| H A D | mkldnn_ir.py | 1023 unary_scalars, argument
|
| H A D | ir.py | 6750 unary_scalars, argument
6948 unary_scalars, argument
7144 unary_scalars, argument
7346 unary_scalars, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 347 torch::List<std::optional<at::Scalar>> unary_scalars, in mkldnn_convolution_pointwise_binary()
511 torch::List<std::optional<at::Scalar>> unary_scalars, in mkldnn_convolution_pointwise_binary_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv.cpp | 1409 torch::List<std::optional<at::Scalar>> unary_scalars, in _quantized_convolution_onednn()
1939 torch::List<std::optional<at::Scalar>> unary_scalars, in run_pointwise_binary()
|