Searched refs:libdevice_dir (Results 1 – 2 of 2) sorted by relevance
121 std::string libdevice_dir = tensorflow::RocdlRoot(); in GetGpuBinaryBlob() local125 llvm_module_copy.get(), gpu_version, config, libdevice_dir); in GetGpuBinaryBlob()164 TF_ASSIGN_OR_RETURN(std::string libdevice_dir, GetLibdeviceDir(config)); in GetGpuBinaryBlob()192 config, libdevice_dir, enable_fusion)); in GetGpuBinaryBlob()238 std::string libdevice_dir = in GetLibdeviceDir() local240 VLOG(2) << "Looking for libdevice at " << libdevice_dir; in GetLibdeviceDir()241 if (tensorflow::Env::Default()->IsDirectory(libdevice_dir).ok()) { in GetLibdeviceDir()242 VLOG(2) << "Found libdevice dir " << libdevice_dir; in GetLibdeviceDir()243 return libdevice_dir; in GetLibdeviceDir()
86 string libdevice_dir = in GetLibdeviceDir() local88 VLOG(2) << "Looking for libdevice at " << libdevice_dir; in GetLibdeviceDir()89 if (tensorflow::Env::Default()->IsDirectory(libdevice_dir).ok()) { in GetLibdeviceDir()90 VLOG(2) << "Found libdevice dir " << libdevice_dir; in GetLibdeviceDir()91 return libdevice_dir; in GetLibdeviceDir()310 std::string libdevice_dir; in CompileTargetBinary() local320 libdevice_dir = cached_libdevice_dir_; in CompileTargetBinary()322 VLOG(2) << "Libdevice dir = " << libdevice_dir << "\n"; in CompileTargetBinary()330 module_config, libdevice_dir)); in CompileTargetBinary()