Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dutils_test.cc38 llvm::LLVMContext llvm_context; in TEST() local
41 tensorflow::GetDataDependencyFilepath(SaxpyIRFile()), &llvm_context); in TEST()
Dutils.cc42 llvm::LLVMContext* llvm_context) { in LoadIRModule() argument
46 diagnostic_err, *llvm_context)); in LoadIRModule()
Dutils.h37 llvm::LLVMContext* llvm_context);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc41 llvm::LLVMContext llvm_context; in CompileAndPrintLlvmIr() local
60 hlo_module.get(), &llvm_context, in CompileAndPrintLlvmIr()
Dmlir_gpu_test_base.cc43 llvm::LLVMContext llvm_context; in CompileMlirModule() local
44 auto llvm_module = absl::make_unique<llvm::Module>("", llvm_context); in CompileMlirModule()
/external/tensorflow/tensorflow/compiler/aot/
Dembedded_protocol_buffers.cc117 llvm::LLVMContext llvm_context; in CreateEmbeddedProtocolBuffers() local
119 absl::make_unique<llvm::Module>("embedded_data_module", llvm_context); in CreateEmbeddedProtocolBuffers()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_to_kernel.cc75 llvm::LLVMContext llvm_context; in EmitToBinary() local
78 mlir::translateModuleToLLVMIR(module, llvm_context); in EmitToBinary()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc549 HloModule* hlo_module, llvm::LLVMContext* llvm_context, in CompileModuleToLlvmIrImpl() argument
559 *llvm_module = absl::make_unique<llvm::Module>("", *llvm_context); in CompileModuleToLlvmIrImpl()
865 llvm::LLVMContext llvm_context; in RunBackend() local
905 module.get(), &llvm_context, target_triple_, data_layout_, in RunBackend()
984 HloModule* hlo_module, llvm::LLVMContext* llvm_context, in CompileModuleToLlvmIr() argument
994 hlo_module, llvm_context, target_triple, data_layout, platform_name, in CompileModuleToLlvmIr()
Dgpu_compiler.h154 HloModule* hlo_module, llvm::LLVMContext* llvm_context,
Dir_emitter_unnested.cc170 llvm::LLVMContext& llvm_context = llvm_module->getContext(); in UpdateLaunchDimensions() local
172 llvm::IntegerType::get(llvm_context, /*NumBits=*/32), in UpdateLaunchDimensions()
177 llvm_context, in UpdateLaunchDimensions()
179 llvm::MDString::get(llvm_context, "reqntidx"), in UpdateLaunchDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc677 auto llvm_context = std::make_unique<llvm::LLVMContext>(); in RunBackend() local
679 absl::make_unique<llvm::Module>("__compute_module", *llvm_context); in RunBackend()
788 std::move(llvm_context)); in RunBackend()
895 llvm::LLVMContext llvm_context; in CompileAheadOfTime() local
896 llvm::Module llvm_module("__compute_module", llvm_context); in CompileAheadOfTime()
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_llvm.h830 draw_llvm_create(struct draw_context *draw, LLVMContextRef llvm_context);