Home
last modified time | relevance | path

Searched refs:cuda_root (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc236 for (const std::string& cuda_root : tensorflow::CandidateCudaRoots( in GetLibdeviceDir() local
239 tensorflow::io::JoinPath(cuda_root, "nvvm", "libdevice"); in GetLibdeviceDir()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc85 for (const string& cuda_root : CandidateCudaRoots(hlo_module_config)) { in GetLibdeviceDir() local
87 tensorflow::io::JoinPath(cuda_root, "nvvm", "libdevice"); in GetLibdeviceDir()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc164 for (const std::string& cuda_root : in findCudaExecutable() local
166 binary_path = tensorflow::io::JoinPath(cuda_root, "bin", binary_filename); in findCudaExecutable()