Searched refs:XRTBufferAllocation (Results 1 – 2 of 2) sorted by relevance
132 XRTBufferAllocation::XRTBufferAllocation(const se::DeviceMemoryBase& allocation, in XRTBufferAllocation() function in tensorflow::XRTBufferAllocation147 XRTBufferAllocation::~XRTBufferAllocation() { in ~XRTBufferAllocation()158 const se::DeviceMemoryBase& XRTBufferAllocation::allocation() { in allocation()162 void XRTBufferAllocation::DiscardAllocation() { in DiscardAllocation()293 typedef XRTBufferAllocation* XRTBufferAllocationPtr;330 new XRTBufferAllocation(se::DeviceMemoryBase(), in MakeSubBuffer()455 *buffer = new XRTBufferAllocation( in MakeTuple()469 new XRTBufferAllocation(tuple_buffers.buffer(index), device_ordinal, in MakeTuple()504 buffer.second = new XRTBufferAllocation(shaped_buffer.buffer(buffer.first), in InitializeFromShapedBuffer()521 XRTBufferAllocation* source_buffer = source.buffers_.element(source_index); in AliasBufferFrom()[all …]
45 class XRTBufferAllocation : public core::RefCounted {47 XRTBufferAllocation(const se::DeviceMemoryBase& allocation,50 ~XRTBufferAllocation() override;227 xla::ShapeTree<XRTBufferAllocation*> buffers_;