Searched refs:preferred_cuda_dir (Results 1 – 4 of 4) sorted by relevance
33 std::string preferred_cuda_dir; member38 absl::string_view preferred_cuda_dir = "",41 preferred_cuda_dir(preferred_cuda_dir), in disable_gpuasm_optimizations()48 return std::make_tuple(disable_gpuasm_optimizations, preferred_cuda_dir, in ToTuple()
154 const std::string preferred_cuda_dir) { in findCudaExecutable() argument165 tensorflow::CandidateCudaRoots(preferred_cuda_dir)) { in findCudaExecutable()202 findCudaExecutable("ptxas", options.preferred_cuda_dir); in CompileGpuAsm()286 std::vector<CubinOrPTXImage> images, const std::string preferred_cuda_dir) { in BundleGpuAsm() argument288 findCudaExecutable("fatbinary", preferred_cuda_dir); in BundleGpuAsm()
66 std::vector<CubinOrPTXImage> images, const std::string preferred_cuda_dir);
225 gpu_asm_opts.preferred_cuda_dir); in GetGpuBinaryBlob()