Home
last modified time | relevance | path

Searched defs:mkl (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/aten/src/ATen/native/mkldnn/
DRegisterMkldnnOpContextClass.cpp109 namespace mkl { namespace
111 TORCH_LIBRARY(mkl, m) { in TORCH_LIBRARY() argument
DLinear.cpp419 TORCH_LIBRARY_IMPL(mkl, CPU, m) { in TORCH_LIBRARY_IMPL() argument
423 TORCH_LIBRARY_IMPL(mkl, MkldnnCPU, m) { in TORCH_LIBRARY_IMPL() argument
DMKLDNNConversions.cpp632 TORCH_LIBRARY_IMPL(mkl, CPU, m) { in TORCH_LIBRARY_IMPL() argument
/external/pytorch/torch/backends/
D__init__.py66 mkl as mkl, unknown
/external/pytorch/aten/src/ATen/native/mkl/
DSparseBlasImpl.cpp31 namespace mkl { namespace
/external/pytorch/aten/src/ATen/mkl/
DSparseBlas.cpp8 namespace at::mkl::sparse { namespace
/external/tensorflow/tensorflow/core/transforms/remapper/
Dpass.cc41 namespace mkl { namespace