Searched refs:IsMklEnabled (Results 1 – 14 of 14) sorted by relevance
45 bool IsMklEnabled();
62 bool IsMklEnabled() { in IsMklEnabled() function
27 m.def("IsMklEnabled", tensorflow::IsMklEnabled); in PYBIND11_MODULE()
108 if ((test_util.IsMklEnabled() and node_op_type in ("_MklAddN", "Mul"))
1379 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1394 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1409 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1424 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1439 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1454 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1469 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1484 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1499 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():1516 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
289 (test_util.IsMklEnabled() and is_eager is False):317 (test_util.IsMklEnabled() and is_eager is False):
749 if test_util.IsMklEnabled():
164 cpuname = 'mklcpu' if test_util.IsMklEnabled() else 'cpu'
33 tensorflow::IsMklEnabled
54 return "_MklMatMul" if test_util.IsMklEnabled() else "MatMul"931 if test_util.IsMklEnabled():
241 if test_util.IsMklEnabled():
327 def IsMklEnabled(): function328 return _pywrap_util_port.IsMklEnabled()
144 if test_util.IsMklEnabled():
373 if mode == 'mkl' and not test_util.IsMklEnabled():