Searched defs:buffer_or (Results 1 – 4 of 4) sorted by relevance
271 StatusOr<std::unique_ptr<PjRtBuffer::ExternalReference>> buffer_or = in BufferToDLPackManagedTensor() local
523 StatusOr<std::shared_ptr<TrackedDeviceBuffer>>&& buffer_or) { in Acquire()1011 StatusOr<std::unique_ptr<PjRtBuffer>> buffer_or = AllocateDestinationBuffer( in MakeCrossHostReceiveBuffers() local1059 StatusOr<std::unique_ptr<PjRtBuffer>> buffer_or = AllocateDestinationBuffer( in MakeCrossHostReceiveBuffersForGather() local
612 StatusOr<std::shared_ptr<TrackedTfrtCpuDeviceBuffer>>&& buffer_or) { in Acquire()
1391 auto buffer_or = BuildBuffer(inst); in BuildSubGraph() local