Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc193 string kernel_name = ir_emitter_context_->name_uniquer()->GetUniqueName( in BuildKernelPrototype()
197 llvm::Module* module = ir_emitter_context_->llvm_module(); in BuildKernelPrototype()
416 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
446 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
468 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
499 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
590 hlo_module_config_, ir_emitter_context_->llvm_module(), &b_, in HandleFusion()
611 hlo_module_config_, ir_emitter_context_->llvm_module(), &b_, in HandleFusion()
644 fusion, ir_emitter_context_->buffer_assignment())) { in HandleFusion()
654 ir_emitter_context_->llvm_module(), in HandleFusion()
[all …]
Dir_emitter_nested.cc69 root_shape, ir_emitter_context_->llvm_module()->getDataLayout()); in EmitBasePointersForNestedComputation()
80 ir_emitter_context_->name_uniquer()->GetUniqueName( in EmitBasePointersForNestedComputation()
83 ir_emitter_context_->llvm_module()); // The parent LLVM module. in EmitBasePointersForNestedComputation()
Dir_emitter.h140 return ir_emitter_context_->buffer_assignment()
173 IrEmitterContext* ir_emitter_context_; variable
Dir_emitter.cc59 : ir_emitter_context_(ir_emitter_context), in IrEmitter()
159 ir_emitter_context_); in EmitCallToNestedComputation()