Searched refs:DLContextForDevice (Results 1 – 1 of 1) sorted by relevance
218 StatusOr<DLContext> DLContextForDevice(const PjRtDevice& device) { in DLContextForDevice() function288 TF_ASSIGN_OR_RETURN(dt.ctx, DLContextForDevice(*buffer->buffer()->device())); in BufferToDLPackManagedTensor()