Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc111 std::string libdevice_dir = tensorflow::RocdlRoot(); in GetGpuBinaryBlob() local
216 std::string libdevice_dir = in GetLibdeviceDir() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc90 string libdevice_dir = in GetLibdeviceDir() local
361 std::string libdevice_dir; in CompileTargetBinary() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc83 std::string libdevice_dir = xla::gpu::GetLibdeviceDir(hlo_module->config()); in CompileAndPrintLlvmIr() local