Searched refs:IsBuiltWithROCm (Results 1 – 8 of 8) sorted by relevance
25 bool IsBuiltWithROCm();
29 bool IsBuiltWithROCm() { in IsBuiltWithROCm() function
22 m.def("IsBuiltWithROCm", tensorflow::IsBuiltWithROCm); in PYBIND11_MODULE()
96 return _test_util.IsBuiltWithROCm()
50 if not test_util.IsBuiltWithROCm() and config.list_physical_devices('GPU'):
30 tensorflow::IsBuiltWithROCm
142 if test_util.IsBuiltWithROCm():
311 def IsBuiltWithROCm(): function312 return _pywrap_util_port.IsBuiltWithROCm()