| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Backend.h | 59 MkldnnCPU, enumerator
113 } else if (t == DispatchKey::MkldnnCPU) { in dispatchKeyToBackend()
114 return Backend::MkldnnCPU; in dispatchKeyToBackend()
191 case Backend::MkldnnCPU: in backendToDispatchKey()
192 return DispatchKey::MkldnnCPU; in backendToDispatchKey()
223 case Backend::MkldnnCPU: in backendToDeviceType()
332 case Backend::MkldnnCPU: in toString()
333 return "MkldnnCPU"; in toString()
|
| H A D | DispatchKey.cpp | 89 case DispatchKey::MkldnnCPU: in toString()
90 return "MkldnnCPU"; in toString()
276 {"MkldnnCPU", c10::DispatchKey::MkldnnCPU}, in parseDispatchKey()
|
| H A D | TensorOptions.h | 695 return DispatchKey::MkldnnCPU; in computeDispatchKey()
737 case DispatchKey::MkldnnCPU: in dispatchKeyToLayout()
755 case DispatchKey::MkldnnCPU: in dispatchKeyToDeviceType()
|
| H A D | Layout.h | 40 case Backend::MkldnnCPU: in layout_from_backend()
|
| H A D | DispatchKeySet.h | 696 constexpr DispatchKeySet mkldnn_ks = DispatchKeySet(DispatchKey::MkldnnCPU);
711 DispatchKey::MkldnnCPU,
|
| H A D | DispatchKey.h | 213 MkldnnCPU, // registered at build/aten/src/ATen/RegisterMkldnnCPU.cpp enumerator
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 41 MkldnnCPU: mkldnn_add_out
51 MkldnnCPU: mkldnn_add
66 MkldnnCPU: mkldnn_mul_out
75 MkldnnCPU: mkldnn_mul
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 554 MkldnnCPU: mkldnn_add
566 MkldnnCPU: mkldnn_add_
582 MkldnnCPU: mkldnn_add_out
1762 MkldnnCPU: copy_mkldnn_
2372 MkldnnCPU: empty_mkldnn
3317 MkldnnCPU: mkldnn_linear
3322 MkldnnCPU: mkldnn_linear_backward_input
3327 MkldnnCPU: mkldnn_linear_backward_weights
3332 MkldnnCPU: mkldnn_linear_backward
3856 MkldnnCPU: mkldnn_max_pool2d
[all …]
|
| H A D | ConvUtils.h | 391 // disable NHWC for MkldnnCPU tensor. in mkldnn_conv_use_channels_last()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear.cpp | 917 at::Tensor onednn_weight, // int8 tensor from MkldnnCPU in linear_int8_with_onednn_weight()
1198 Tensor onednn_weight, // int8 tensor from MkldnnCPU in run_pointwise()
1227 Tensor onednn_weight, // int8 tensor from MkldnnCPU in run_pointwise_tensor()
1258 Tensor onednn_weight, // int8 tensor from MkldnnCPU in run_pointwise_binary()
1290 Tensor onednn_weight, // int8 tensor from MkldnnCPU in run_pointwise_binary_tensor()
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/aten/src/ATen/native/mkldnn/ |
| H A D | MKLDNNCommon.cpp | 100 DispatchKeySet(DispatchKey::MkldnnCPU), in new_with_itensor_mkldnn()
203 TORCH_LIBRARY_IMPL(mkldnn, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
|
| 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 | 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/torch/csrc/jit/codegen/onednn/ |
| H A D | LlgaTensorImpl.cpp | 52 c10::DispatchKeySet(c10::DispatchKey::MkldnnCPU), in LlgaTensorImpl()
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 92 MkldnnCPU = auto() variable in DispatchKey
278 DispatchKey.MkldnnCPU,
|