Searched defs:mkldnn (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RegisterMkldnnOpContextClass.cpp | 14 namespace mkldnn { namespace
34 TORCH_LIBRARY(mkldnn, m) { in TORCH_LIBRARY() argument
|
| H A D | Linear.cpp | 429 TORCH_LIBRARY_IMPL(mkldnn, CPU, m) { in TORCH_LIBRARY_IMPL() argument
438 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | MKLDNNConversions.cpp | 541 TORCH_LIBRARY_IMPL(mkldnn, CPU, m) { in TORCH_LIBRARY_IMPL() argument
556 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | IDeepRegistration.cpp | 22 namespace at::native::mkldnn{ namespace
|
| H A D | Conv.cpp | 1048 TORCH_LIBRARY_IMPL(mkldnn, CPU, m) { in TORCH_LIBRARY_IMPL() argument
1063 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
1078 TORCH_LIBRARY_IMPL(mkldnn, Meta, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | OpContext.cpp | 8 namespace mkldnn { namespace
|
| H A D | MKLDNNCommon.cpp | 203 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | ConvPrepack.cpp | 16 namespace mkldnn { namespace
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ParallelOpenMP.cpp | 17 namespace native { namespace mkldnn { namespace
|
| /aosp_15_r20/external/pytorch/torch/backends/ |
| H A D | __init__.py | 67 mkldnn as mkldnn, unknown
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_custom_post_grad_passes.py | 54 mkldnn = torch.ops.mkldnn variable
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | mkldnn_fusion.py | 32 mkldnn = torch.ops.mkldnn variable
|