Home
last modified time | relevance | path

Searched defs:llvm_module (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h44 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext()
69 llvm::Module* llvm_module() { return llvm_module_; } in llvm_module() function
Dhlo_to_ir_bindings.h36 HloToIrBindings(llvm::IRBuilder<>* b, llvm::Module* llvm_module, in HloToIrBindings()
Dnvptx_compiler.cc259 llvm::Module* llvm_module) { in MaybeLoadLLVMFromFile()
356 llvm::Module* llvm_module, in CompileTargetBinary()
Damdgpu_compiler.cc129 llvm::Module* llvm_module, in CompileTargetBinary()
Dgpu_compiler.cc730 std::unique_ptr<llvm::Module> llvm_module; member
844 std::unique_ptr<llvm::Module> llvm_module, in CompileToTargetBinary()
853 absl::optional<int> shard_number) -> StatusOr<BackendCompileResult> { in CompileToTargetBinary()
1292 std::unique_ptr<llvm::Module> llvm_module, in CompileLmhloToExecutable()
Dir_emitter_unnested.cc160 llvm::Module* llvm_module) { in AnnotateThunkLaunchDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc30 llvm::Module* llvm_module, const int64_t num_dynamic_loop_bounds) { in GetComputeFunctionParams()
47 llvm::Module* llvm_module, llvm::IRBuilder<>* b, in IrFunction()
Dcpu_compiler.cc531 const llvm::Module& llvm_module) { in GetIRModuleHooks()
547 Status VerifyLlvmModule(const llvm::Module& llvm_module) { in VerifyLlvmModule()
702 auto llvm_module = in RunBackend() local
930 llvm::Module llvm_module("__compute_module", llvm_context); in CompileAheadOfTime() local
Dmlir_emitter.cc95 llvm::Module *llvm_module = b->GetInsertBlock()->getParent()->getParent(); in EmitMlirFuncAndCall() local
Dir_emitter.cc93 const BufferAssignment& assignment, llvm::Module* llvm_module, in IrEmitter()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_to_kernel.cc77 std::unique_ptr<llvm::Module> llvm_module = in EmitToBinary() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc45 auto llvm_module = absl::make_unique<llvm::Module>("", llvm_context); in CompileMlirModule() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc61 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.cc584 const llvm::Module& llvm_module, bool optimized, in DumpIrIfEnabled()
/external/mesa3d/src/amd/vulkan/
Dradv_nir_to_llvm.c4125 LLVMModuleRef llvm_module, in ac_compile_llvm_module()
4182 LLVMModuleRef llvm_module; in radv_compile_nir_shader() local