Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc63 temp_buffer_base_ = &*arg_iter; in EmitBasePointersForHlos()
64 temp_buffer_base_->setName("temp_buffer"); in EmitBasePointersForHlos()
82 CHECK_NE(nullptr, temp_buffer_base_); in EmitBasePointersForHlos()
85 b_->CreateInBoundsGEP(temp_buffer_base_, b_->getInt64(offset)); in EmitBasePointersForHlos()
122 CHECK_NE(nullptr, temp_buffer_base_); in EmitBasePointersForHlos()
125 b_->CreateInBoundsGEP(temp_buffer_base_, b_->getInt64(offset)), in EmitBasePointersForHlos()
256 " temp_buffer_base_=", llvm_ir::DumpToString(*temp_buffer_base_), in ToString()
Dhlo_to_ir_bindings.h68 llvm::Value* GetTempBufferBase() const { return temp_buffer_base_; } in GetTempBufferBase()
69 void SetTempBufferBase(llvm::Value* v) { temp_buffer_base_ = v; } in SetTempBufferBase()
118 llvm::Value* temp_buffer_base_ = nullptr; variable
Dbuffer_allocations.h79 se::DeviceMemoryBase GetTempBufferBase() const { return temp_buffer_base_; } in GetTempBufferBase()
104 se::DeviceMemoryBase temp_buffer_base_; variable
Dbuffer_allocations.cc103 buffer_allocations->temp_buffer_base_ = buffer_address; in Build()