Searched refs:InitializeFromShapedBuffer (Results 1 – 2 of 2) sorted by relevance
191 ->InitializeFromShapedBuffer(shaped_buffer, backend->memory_allocator(), in CreateAndTransfer()214 ->InitializeFromShapedBuffer(shaped_buffer, backend->memory_allocator(), in CreateUninitialized()229 ->InitializeFromShapedBuffer(shaped_buffer, allocator, device_ordinal); in CreateFromBuffer()318 InitializeFromShapedBuffer(shaped_buffer, backend->memory_allocator(), in SwapIn()573 void XRTTupleAllocation::InitializeFromShapedBuffer( in InitializeFromShapedBuffer() function in tensorflow::XRTTupleAllocation
243 void InitializeFromShapedBuffer(const xla::ShapedBuffer& shaped_buffer,