Home
last modified time | relevance | path

Searched refs:CudaSupportsHalfMatMulAndConv (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dport.h26 bool CudaSupportsHalfMatMulAndConv();
Dport.cc29 bool CudaSupportsHalfMatMulAndConv() { in CudaSupportsHalfMatMulAndConv() function
/external/tensorflow/tensorflow/python/util/
Dport.i25 %unignore tensorflow::CudaSupportsHalfMatMulAndConv;
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatmul_op_test.py77 not test_util.CudaSupportsHalfMatMulAndConv()):
Dconv_ops_3d_test.py54 if not test_util.CudaSupportsHalfMatMulAndConv():
Dpooling_ops_test.py212 if not use_gpu or test_util.CudaSupportsHalfMatMulAndConv():
Dconv_ops_test.py166 if use_gpu and not test_util.CudaSupportsHalfMatMulAndConv():
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py255 def CudaSupportsHalfMatMulAndConv(): function
256 return pywrap_tensorflow.CudaSupportsHalfMatMulAndConv()