Searched refs:XRTBufferAllocation (Results 1 – 2 of 2) sorted by relevance
51 class XRTBufferAllocation : public core::RefCounted {53 XRTBufferAllocation(const se::DeviceMemoryBase& allocation,55 ~XRTBufferAllocation() override;287 xla::ShapeTree<XRTBufferAllocation*> buffers_;
37 typedef XRTBufferAllocation* XRTBufferAllocationPtr;119 XRTBufferAllocation::XRTBufferAllocation(const se::DeviceMemoryBase& allocation, in XRTBufferAllocation() function in tensorflow::XRTBufferAllocation133 XRTBufferAllocation::~XRTBufferAllocation() { in ~XRTBufferAllocation()143 const se::DeviceMemoryBase& XRTBufferAllocation::allocation() { in allocation()541 new XRTBufferAllocation(tuple_buffers.buffer(index), device_ordinal, in MakeTuple()581 index_buffer.second = new XRTBufferAllocation( in InitializeFromShapedBuffer()606 XRTBufferAllocation* source_buffer = source.buffers_.element(source_index); in AliasBufferFrom()607 XRTBufferAllocation* dest_buffer = buffers_.element(dest_index); in AliasBufferFrom()