Home
last modified time | relevance | path

Searched refs:setUserEnabledMkldnn (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/test/
Dcpu_profiling_allocator_test.cpp205 at::globalContext().setUserEnabledMkldnn(false); in main()
/external/pytorch/aten/src/ATen/
DContext.cpp47 void Context::setUserEnabledMkldnn(bool e) { in setUserEnabledMkldnn() function in at::Context
DContext.h184 void setUserEnabledMkldnn(bool e);
/external/pytorch/aten/src/ATen/native/
DLinearAlgebra.cpp1521 at::globalContext().setUserEnabledMkldnn(false); in addmm_impl_cpu_()
1776 at::globalContext().setUserEnabledMkldnn(false); in bmm_out_or_baddbmm_()
/external/pytorch/torch/csrc/
DModule.cpp824 at::globalContext().setUserEnabledMkldnn(arg == Py_True); in THPModule_setUserEnabledMkldnn()