Searched defs:MkldnnCPU (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Linear.cpp | 423 TORCH_LIBRARY_IMPL(mkl, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
438 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | MKLDNNCommon.cpp | 203 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | MKLDNNConversions.cpp | 556 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | Conv.cpp | 1063 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Backend.h | 59 MkldnnCPU, enumerator
|
| H A D | DispatchKey.h | 213 MkldnnCPU, // registered at build/aten/src/ATen/RegisterMkldnnCPU.cpp enumerator
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear.cpp | 1339 TORCH_LIBRARY_IMPL(onednn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | qconv.cpp | 2004 TORCH_LIBRARY_IMPL(onednn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 92 MkldnnCPU = auto() variable in DispatchKey
|