Home
last modified time | relevance | path

Searched refs:IsMklEnabled (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dnode_file_writer_test.py30 from tensorflow.python.framework.test_util import IsMklEnabled
141 if not IsMklEnabled():
162 if not IsMklEnabled():
Dconfig_test.py618 if test_util.IsMklEnabled():
679 if test_util.IsMklEnabled():
Dtest_util.py360 def IsMklEnabled(): function
361 return (_pywrap_util_port.IsMklEnabled() or
Dtest_util_test.py145 if test_util.IsMklEnabled():
/external/tensorflow/tensorflow/core/util/
Dport.h45 bool IsMklEnabled();
Dport.cc62 bool IsMklEnabled() { in IsMklEnabled() function
/external/tensorflow/tensorflow/python/util/
Dport_wrapper.cc27 m.def("IsMklEnabled", tensorflow::IsMklEnabled); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dconv_ops_test.py164 if test_util.IsMklEnabled():
846 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1408 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1423 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1438 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1453 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1468 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1483 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1498 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
1513 if test.is_gpu_available(cuda_only=True) or test_util.IsMklEnabled():
[all …]
Dconv_ops_3d_test.py286 (test_util.IsMklEnabled() and is_eager is False):
314 (test_util.IsMklEnabled() and is_eager is False):
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_grappler_test.py104 if ((test_util.IsMklEnabled() and node_op_type in ("_MklAddN", "Mul"))
/external/tensorflow/tensorflow/python/eager/
Dcontext_test.py137 if test_util.IsMklEnabled():
/external/tensorflow/tensorflow/python/grappler/
Dcost_analyzer_test.py119 if not test_util.IsMklEnabled():
Dremapper_test.py98 if mode == 'mkl' and not test_util.IsMklEnabled():
Dauto_mixed_precision_test.py377 if mode == 'mkl' and not test_util.IsMklEnabled():
/external/tensorflow/tensorflow/python/client/
Dtimeline_test.py157 cpuname = 'mklcpu' if test_util.IsMklEnabled() else 'cpu'
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt33 tensorflow::IsMklEnabled
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli_test.py50 if (test_util.IsMklEnabled() and
943 if test_util.IsMklEnabled():
/external/tensorflow/tensorflow/python/profiler/
Dmodel_analyzer_test.py240 if test_util.IsMklEnabled():
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dconcat_op_test.py757 if test_util.IsMklEnabled():