Home
last modified time | relevance | path

Searched refs:CandidateCudaRoots (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/platform/
Dcuda_libdevice_path.h26 std::vector<string> CandidateCudaRoots();
31 inline std::vector<string> CandidateCudaRoots(
33 std::vector<string> candidates = CandidateCudaRoots();
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc63 static std::vector<std::string> CandidateCudaRoots( in CandidateCudaRoots() function
65 return tensorflow::CandidateCudaRoots( in CandidateCudaRoots()
74 for (const auto& dir : CandidateCudaRoots(hlo_module_config)) { in PrintCantFindCudaMessage()
85 for (const string& cuda_root : CandidateCudaRoots(hlo_module_config)) { in GetLibdeviceDir()
/external/tensorflow/tensorflow/core/platform/default/
Dcuda_libdevice_path.cc28 std::vector<string> CandidateCudaRoots() { in CandidateCudaRoots() function
/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()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc165 tensorflow::CandidateCudaRoots(preferred_cuda_dir)) { in findCudaExecutable()