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