Searched refs:ir_emitter_context_ (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_nested.cc | 77 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()
|
D | ir_emitter_unnested.cc | 546 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 …]
|
D | ir_emitter.cc | 81 : 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()
|
D | ir_emitter_unnested.h | 124 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()
|
D | ir_emitter.h | 169 IrEmitterContext* ir_emitter_context_; variable
|