| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_compiler.cc | 934 llvm::Module* llvm_module, in RemoveUnusedAndUninitializedGlobals()
952 std::unique_ptr<llvm::Module> llvm_module; member
1099 std::unique_ptr<llvm::Module> llvm_module, in CompileToTargetBinary()
1108 std::optional<int> shard_number) -> StatusOr<BackendCompileResult> { in CompileToTargetBinary()
1614 std::unique_ptr<llvm::Module> llvm_module, in CompileLmhloToExecutable()
|
| H A D | ir_emitter_context.h | 44 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext()
71 llvm::Module* llvm_module() { return llvm_module_; } in llvm_module() function
|
| H A D | nvptx_compiler.cc | 232 llvm::Module* llvm_module) { in MaybeLoadLLVMFromFile()
329 llvm::Module* llvm_module, in CompileTargetBinary()
|
| H A D | hlo_to_ir_bindings.h | 34 HloToIrBindings(llvm::IRBuilder<>* b, llvm::Module* llvm_module, in HloToIrBindings()
|
| H A D | amdgpu_compiler.cc | 135 llvm::Module* llvm_module, in CompileTargetBinary()
|
| H A D | ir_emitter_unnested.cc | 158 llvm::Module* llvm_module) { in AnnotateWithInt32Value()
176 llvm::Module* llvm_module) { in AnnotateThunkLaunchDimensions()
1601 std::unique_ptr<llvm::Module> llvm_module = mlir::translateModuleToLLVMIR( in EmitLaunchFunc() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_compiler.cc | 751 const llvm::Module& llvm_module) { in GetIRModuleHooks()
767 Status VerifyLlvmModule(const llvm::Module& llvm_module) { in VerifyLlvmModule()
1128 auto llvm_module = in CompileLegacyCpuExecutable() local
1409 std::unique_ptr<llvm::Module> llvm_module; in CompileAheadOfTime() local
|
| H A D | ir_function.cc | 30 llvm::Module* llvm_module, const int64_t num_dynamic_loop_bounds) { in GetComputeFunctionParams()
48 llvm::Module* llvm_module, llvm::IRBuilder<>* b, in IrFunction()
|
| H A D | mlir_emitter.cc | 98 llvm::Module *llvm_module = b->GetInsertBlock()->getParent()->getParent(); in EmitMlirFuncAndCall() local
|
| H A D | ir_emitter.cc | 98 llvm::Module* llvm_module, in IrEmitter()
|
| /aosp_15_r20/external/mesa3d/src/amd/vulkan/ |
| H A D | radv_nir_to_llvm.c | 433 ac_compile_llvm_module(struct ac_llvm_compiler *ac_llvm, LLVMModuleRef llvm_module, struct radv_sha… in ac_compile_llvm_module()
484 LLVMModuleRef llvm_module; in radv_compile_nir_shader() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | kernel_ops_pattern.cc | 220 const xla::HloModuleConfig& hlo_module_config, llvm::Module* llvm_module) { in Emit()
281 auto llvm_module = std::make_unique<llvm::Module>("", llvm_context); in Match() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | tf_to_kernel.cc | 77 std::unique_ptr<llvm::Module> llvm_module = in EmitToBinary() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | mlir_gpu_test_base.cc | 48 auto llvm_module = std::make_unique<llvm::Module>("", llvm_context); in CompileMlirModule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | llvm_compiler_test.cc | 63 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| H A D | jit_compiler.cc | 255 auto llvm_module = translateModuleToLLVMIR(compiler->module(), *llvm_ctx); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | llvm_util.cc | 585 const llvm::Module& llvm_module, bool optimized, in DumpIrIfEnabled()
|