Searched refs:GetTempBufferBase (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ | ||
H A D | buffer_allocations.h | 76 perftools::gputools::DeviceMemoryBase GetTempBufferBase() const { function in class:xla::gpu::BufferAllocations |
H A D | hlo_to_ir_bindings.h | 68 llvm::Value* GetTempBufferBase() const { return temp_buffer_base_; } function in class:xla::gpu::HloToIrBindings |
H A D | ir_emitter.cc | 172 arguments.push_back(bindings_.GetTempBufferBase()); |
Completed in 364 milliseconds