Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Drecording_tpu_driver.cc73 class RecordingBufferHandle : public BufferHandle { class
75 explicit RecordingBufferHandle(std::unique_ptr<BufferHandle> handle) in RecordingBufferHandle() function in tpu_driver::__anone63593b20111::RecordingBufferHandle
166 std::make_unique<RecordingBufferHandle>(std::move(handle)); in Allocate()
189 std::make_unique<RecordingBufferHandle>(std::move(handle)); in Allocate()
214 static_cast<const RecordingBufferHandle*>(child)->handle_.get(); in AllocateTuple()
217 static_cast<const RecordingBufferHandle*>(child)->id_); in AllocateTuple()
224 std::make_unique<RecordingBufferHandle>(std::move(handle)); in AllocateTuple()
248 auto recording_handle = static_cast<RecordingBufferHandle*>(handle.get()); in Deallocate()
271 auto recording_handle = static_cast<RecordingBufferHandle*>(dst); in TransferToDevice()
275 src, static_cast<RecordingBufferHandle*>(dst)->handle_.get(), in TransferToDevice()
[all …]