Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnvptx_helper.cc47 for (const std::string& cuda_root : CandidateCudaRoots(hlo_module_config)) { in GetLibdeviceDir() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dgpu_kernel_to_blob_pass.cc241 for (const std::string& cuda_root : tensorflow::CandidateCudaRoots( in GetLibdeviceDir() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dasm_compiler.cc199 for (const std::string& cuda_root : in FindCudaExecutable() local