Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dgpu_kernel_to_blob_pass.cc85 auto llvmModule = translateModuleToLLVMIR(gpu_module, llvmContext); in GetGpuBinaryBlob() local
86 if (!llvmModule) { in GetGpuBinaryBlob()
90 llvmModule->setModuleIdentifier(gpu_module.getName()); in GetGpuBinaryBlob()
110 auto llvm_module_copy = llvm::CloneModule(*llvmModule); in GetGpuBinaryBlob()
140 llvmModule->setDataLayout(xla::gpu::nvptx::DataLayout()); in GetGpuBinaryBlob()
141 llvmModule->setTargetTriple(xla::gpu::nvptx::TargetTriple()); in GetGpuBinaryBlob()
157 auto llvm_module_copy = llvm::CloneModule(*llvmModule); in GetGpuBinaryBlob()