Searched refs:AllocateDestinationBuffer (Results 1 – 1 of 1) sorted by relevance
363 StatusOr<std::unique_ptr<PjRtStreamExecutorBuffer>> AllocateDestinationBuffer( in AllocateDestinationBuffer() function722 AllocateDestinationBuffer(compact_shape, device, local_device, in BufferFromHostBuffer()851 AllocateDestinationBuffer(compact_shape, device, local_device, in CreateUninitializedBuffer()875 AllocateDestinationBuffer(compact_shape, device, local_device, in BufferFromHostLiteral()942 StatusOr<std::unique_ptr<PjRtBuffer>> buffer_or = AllocateDestinationBuffer( in MakeCrossHostReceiveBuffers()1298 AllocateDestinationBuffer( in CopyToDeviceHelper()