Home
last modified time | relevance | path

Searched refs:ir_emitter_context_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc77 root_shape, ir_emitter_context_->llvm_module()->getDataLayout()); in CodegenNestedComputation()
86 ir_emitter_context_->name_uniquer()->GetUniqueName( in CodegenNestedComputation()
89 ir_emitter_context_->llvm_module()); // The parent LLVM module. in CodegenNestedComputation()
Dir_emitter_unnested.cc546 emitter->ir_emitter_context_->buffer_assignment(), *hlo_computation, in Create()
562 string kernel_name = ir_emitter_context_->name_uniquer()->GetUniqueName( in BuildKernelPrototype()
566 llvm::Module* module = ir_emitter_context_->llvm_module(); in BuildKernelPrototype()
626 v, ir_emitter_context_->allocations()); in GetAllocationSliceForMlir()
763 ir_emitter_context_->llvm_module()->getContext(), content); in EmitConstant()
779 llvm_ir::GetGlobalMemoryAddressSpace(*ir_emitter_context_->llvm_module()); in EmitConstant()
789 ir_emitter_context_->llvm_module()->getGlobalList().push_back( in EmitConstant()
796 ir_emitter_context_->constants().push_back(std::move(info)); in EmitConstant()
925 input_shape, ir_emitter_context_->gpu_device_info(), unroll_factor); in EmitPadToStaticFromMlir()
927 ir_emitter_context_->llvm_module()); in EmitPadToStaticFromMlir()
[all …]
Dir_emitter.cc81 : ir_emitter_context_(ir_emitter_context), in IrEmitter()
132 *ir_emitter_context_->llvm_module()); in EmitConstants()
144 ir_emitter_context_->llvm_module()->getGlobalList().push_back( in EmitConstants()
156 ir_emitter_context_->buffer_assignment().GetUniqueSlice(instr, {}); in EmitConstants()
161 ir_emitter_context_->constants().push_back(std::move(info)); in EmitConstants()
234 ir_emitter_context_)); in EmitCallToNestedComputation()
295 ir_emitter_context_->cuda_compute_capability()->cc_major >= 6; in MaybeEmitDirectAtomicOperation()
Dir_emitter_unnested.h124 return ir_emitter_context_->platform_name(); in platform_name()
344 return ir_emitter_context_->buffer_assignment().GetUniqueSlice(&hlo, index); in MaybeGetAllocationSlice()
356 shape, ir_emitter_context_->llvm_module()->getDataLayout()); in ByteSizeOf()
Dir_emitter.h169 IrEmitterContext* ir_emitter_context_; variable