Searched refs:userEnabledMkldnn (Results 1 – 9 of 9) sorted by relevance
100 return at::globalContext().userEnabledMkldnn() && mkldnn_bf16_device_check(); in use_mkldnn_bf16_matmul()104 return at::globalContext().userEnabledMkldnn() && mkldnn_fp16_device_check(); in use_mkldnn_fp16_matmul()
43 bool Context::userEnabledMkldnn() const { in userEnabledMkldnn() function in at::Context
183 bool userEnabledMkldnn() const;
1078 if (!at::globalContext().userEnabledMkldnn()) { in device_check()1117 if (!at::globalContext().userEnabledMkldnn()) { in need_pack()
376 if (!at::globalContext().userEnabledMkldnn()) { in use_mkldnn()
1389 …return at::globalContext().userEnabledMkldnn() && m > min_dim && k > min_dim && n > min_dim && m *… in apply_mkldnn_matmul_heur()3559 if (at::globalContext().userEnabledMkldnn()) { in _int_mm_out_cpu()
88 if (!at::globalContext().userEnabledMkldnn()) { in use_mkldnn()
518 if (!at::globalContext().userEnabledMkldnn()) { in use_mkldnn()
830 if (at::globalContext().userEnabledMkldnn()) in THPModule_userEnabledMkldnn()