Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dgpu_kernel_to_blob_pass.cc109 std::string libdevice_dir = tensorflow::RocdlRoot(); in GetGpuBinaryBlob() local
243 std::string libdevice_dir = in GetLibdeviceDir() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dkernel_ops_pattern.cc310 auto libdevice_dir = tensorflow::RocdlRoot(); in Match() local
317 auto libdevice_dir = xla::gpu::GetLibdeviceDir(hlo_module_config); in Match() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnvptx_helper.cc48 std::string libdevice_dir = in GetLibdeviceDir() local
H A Dnvptx_compiler.cc334 std::string libdevice_dir; in CompileTargetBinary() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dhlo_to_llvm_ir.cc90 std::string libdevice_dir = xla::gpu::GetLibdeviceDir(hlo_module->config()); in CompileAndPrintLlvmIr() local