Searched refs:GpuSupportsHalfMatMulAndConv (Results 1 – 9 of 9) sorted by relevance
42 bool GpuSupportsHalfMatMulAndConv();
53 bool GpuSupportsHalfMatMulAndConv() { in GpuSupportsHalfMatMulAndConv() function
26 tensorflow::GpuSupportsHalfMatMulAndConv); in PYBIND11_MODULE()
79 not test_util.GpuSupportsHalfMatMulAndConv()):
62 if not test_util.GpuSupportsHalfMatMulAndConv():
239 if not use_gpu or test_util.GpuSupportsHalfMatMulAndConv():
170 if use_gpu and not test_util.GpuSupportsHalfMatMulAndConv():
32 tensorflow::GpuSupportsHalfMatMulAndConv
323 def GpuSupportsHalfMatMulAndConv(): function324 return _pywrap_util_port.GpuSupportsHalfMatMulAndConv()