Home
last modified time | relevance | path

Searched refs:libdevice_dir_path (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dnvptx_backend_lib.cc74 static string GetLibdeviceFilename(const string& libdevice_dir_path, in GetLibdeviceFilename() argument
80 tensorflow::io::JoinPath(libdevice_dir_path, unified_libdevice_filename), in GetLibdeviceFilename()
302 const string& libdevice_dir_path) { in LinkLibdeviceIfNecessary() argument
309 libdevice_dir_path, in LinkLibdeviceIfNecessary()
310 GetLibdeviceFilename(libdevice_dir_path, compute_capability)); in LinkLibdeviceIfNecessary()
331 const string& libdevice_dir_path) { in CompileModuleToPtx() argument
342 LinkLibdeviceIfNecessary(module, compute_capability, libdevice_dir_path)); in CompileModuleToPtx()
485 const string& libdevice_dir_path) { in CompileToPtx() argument
497 libdevice_dir_path)); in CompileToPtx()
Dnvptx_backend_lib.h42 const string& libdevice_dir_path);