| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_context.h | 36 llvm::Module* llvm_module)
40 llvm_module_(llvm_module) {}
53 llvm::Module* llvm_module() { return llvm_module_; } function in class:xla::gpu::IrEmitterContext
|
| hlo_to_ir_bindings.h | 39 llvm::IRBuilder<>* ir_builder, llvm::Module* llvm_module,
44 module_(llvm_module),
|
| gpu_compiler.cc | 500 llvm::Module llvm_module(module->name().c_str(), llvm_context);
502 llvm_module.setTargetTriple(kTargetTriple);
503 llvm_module.setDataLayout(kDataLayout);
537 &llvm_module);
549 TF_CHECK_OK(user_pre_optimization_hook_(llvm_module));
555 ir_module_string_before_opt = llvm_ir::DumpModuleToString(llvm_module);
566 /*hlo_module_name=*/module->name(), llvm_module,
577 TF_RET_CHECK(!llvm::verifyModule(llvm_module, &err_stream))
612 TF_ASSIGN_OR_RETURN(ptx, CompileToPtx(&llvm_module, {cc_major, cc_minor},
619 /*hlo_module_name=*/module->name(), llvm_module,
[all...] |
| ir_emitter_nested.cc | 68 root_shape, ir_emitter_context_->llvm_module()->getDataLayout());
82 ir_emitter_context_->llvm_module()); // The parent LLVM module.
|
| ir_emitter_unnested.cc | 118 // of the corresponding IR kernel in "llvm_module".
121 llvm::Module* llvm_module) {
129 llvm_module->getOrInsertNamedMetadata("nvvm.annotations");
131 llvm_module->getFunction(kernel_thunk->kernel_name().c_str());
132 llvm::LLVMContext& llvm_context = llvm_module->getContext();
235 llvm::Module* module = ir_emitter_context_->llvm_module();
514 hlo_module_config_, ir_emitter_context_->llvm_module(),
576 ir_emitter_context_->llvm_module(),
591 ir_emitter_context_->llvm_module());
[all...] |
| ir_emitter.cc | 58 module_(ir_emitter_context->llvm_module()),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_compiler.cc | 396 hlo_module_name](const llvm::Module& llvm_module) {
398 TF_RETURN_IF_ERROR(user_pre_optimization_hook(llvm_module));
402 llvm_module,
408 hlo_module_name](const llvm::Module& llvm_module) {
410 TF_RETURN_IF_ERROR(user_post_optimization_hook(llvm_module));
414 llvm_module,
421 Status VerifyLlvmModule(const llvm::Module& llvm_module) {
428 TF_RET_CHECK(!llvm::verifyModule(llvm_module, &err_stream))
473 auto llvm_module = local
483 llvm_module->setDataLayout(jit->data_layout())
[all...] |
| ir_function.cc | 34 llvm::Module* llvm_module, const int64 num_dynamic_loop_bounds) {
35 llvm::Type* i8_ptr_type = llvm::Type::getInt8PtrTy(llvm_module->getContext());
38 llvm::Type::getInt64PtrTy(llvm_module->getContext());
51 const bool enable_fast_math, llvm::Module* llvm_module,
55 llvm_module_(llvm_module),
|
| ir_function.h | 56 const bool enable_fast_math, llvm::Module* llvm_module,
|
| ir_emitter.h | 65 // llvm_module: the LLVM module to emit IR into.
74 llvm::Module* llvm_module,
|
| ir_emitter.cc | 83 llvm::Module* llvm_module,
89 module_(llvm_module),
90 arch_type_(llvm::Triple(llvm_module->getTargetTriple()).getArch()),
91 ir_builder_(llvm_module->getContext()),
94 alias_analysis_(hlo_module, assignment, &llvm_module->getContext()),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| llvm_util.h | 276 // Dumps out `llvm_module` to a file in the directory named `directory_name`,
283 const llvm::Module& llvm_module, bool optimized);
|
| llvm_util.cc | 702 const llvm::Module& llvm_module, bool optimized) {
721 directory_name, ir_file_name, DumpModuleToString(llvm_module)));
724 DumpModuleToString(*DropConstantInitializers(llvm_module)));
|
| /external/mesa3d/src/amd/common/ |
| ac_nir_to_llvm.c | 4675 LLVMModuleRef llvm_module = ac_translate_nir_to_llvm(tm, nir, shader_info, local
[all...] |
