Searched refs:rocm_root_dir (Results 1 – 2 of 2) sorted by relevance
360 const std::string& rocm_root_dir) { in findRocmExecutable() argument363 tensorflow::io::JoinPath(rocm_root_dir, binary_relative_path); in findRocmExecutable()364 VLOG(2) << "Looking for " << binary_relative_path << " at " << rocm_root_dir; in findRocmExecutable()372 std::vector<HsacoImage> images, const std::string rocm_root_dir) { in BundleGpuAsm() argument374 findRocmExecutable("llvm/bin/clang-offload-bundler", rocm_root_dir); in BundleGpuAsm()
76 std::vector<HsacoImage> images, const std::string rocm_root_dir);