Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h45 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext()
70 llvm::Module* llvm_module() { return llvm_module_; } in llvm_module() function
Dgpu_compiler.cc555 std::unique_ptr<llvm::Module>* llvm_module, in CompileModuleToLlvmIrImpl()
677 std::unique_ptr<llvm::Module> llvm_module, in CompileToTargetBinary()
686 absl::optional<int> shard_number) -> StatusOr<BackendCompileResult> { in CompileToTargetBinary()
899 std::unique_ptr<llvm::Module> llvm_module; in RunBackend() local
989 std::unique_ptr<llvm::Module> llvm_module; in CompileModuleToLlvmIr() local
1101 std::unique_ptr<llvm::Module> llvm_module, in CompileLmhloToExecutable()
Dhlo_to_ir_bindings.h39 llvm::IRBuilder<>* b, llvm::Module* llvm_module, in HloToIrBindings()
Damdgpu_compiler.cc115 llvm::Module* llvm_module, in CompileTargetBinary()
Dnvptx_compiler.cc302 llvm::Module* llvm_module, in CompileTargetBinary()
Dir_emitter_unnested.cc159 llvm::Module* llvm_module) { in UpdateLaunchDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc30 llvm::Module* llvm_module, const int64 num_dynamic_loop_bounds) { in GetComputeFunctionParams()
47 llvm::Module* llvm_module, llvm::IRBuilder<>* b, in IrFunction()
Dcpu_compiler.cc506 const llvm::Module& llvm_module) { in GetIRModuleHooks()
522 Status VerifyLlvmModule(const llvm::Module& llvm_module) { in VerifyLlvmModule()
678 auto llvm_module = in RunBackend() local
896 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.cc44 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/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/
DIRForTarget.cpp1841 bool IRForTarget::runOnModule(Module &llvm_module) { in runOnModule()
/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