Searched refs:buffer_or (Results 1 – 4 of 4) sorted by relevance
264 StatusOr<std::unique_ptr<PjRtBuffer::ExternalReference>> buffer_or = in BufferToDLPackManagedTensor() local267 if (!buffer_or.ok()) { in BufferToDLPackManagedTensor()270 buffer_or.status().ToString()); in BufferToDLPackManagedTensor()272 pack->external_reference = buffer_or.ConsumeValueOrDie(); in BufferToDLPackManagedTensor()
523 StatusOr<std::shared_ptr<TrackedDeviceBuffer>>&& buffer_or) { in Acquire() argument525 if (buffer_or.ok()) { in Acquire()526 buffer_ = buffer_or.ValueOrDie(); in Acquire()529 status_ = buffer_or.status(); in Acquire()942 StatusOr<std::unique_ptr<PjRtBuffer>> buffer_or = AllocateDestinationBuffer( in MakeCrossHostReceiveBuffers() local946 if (!buffer_or.ok()) { in MakeCrossHostReceiveBuffers()947 notifier(buffer_or.status()); in MakeCrossHostReceiveBuffers()950 buffers.push_back(buffer_or.ConsumeValueOrDie()); in MakeCrossHostReceiveBuffers()
442 void Acquire(StatusOr<std::shared_ptr<TrackedDeviceBuffer>>&& buffer_or);
1372 auto buffer_or = BuildBuffer(inst); in BuildSubGraph() local1373 if (!buffer_or) return false; in BuildSubGraph()1374 buffers_.push_back(*buffer_or); in BuildSubGraph()