From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTP id EA658495D7 for ; Tue, 13 Feb 2024 22:49:10 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 30F2A68D13C; Wed, 14 Feb 2024 00:49:08 +0200 (EET) Received: from wfhigh6-smtp.messagingengine.com (wfhigh6-smtp.messagingengine.com [64.147.123.157]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 1A3A968CAAB for ; Wed, 14 Feb 2024 00:49:02 +0200 (EET) Received: from compute3.internal (compute3.nyi.internal [10.202.2.43]) by mailfhigh.west.internal (Postfix) with ESMTP id 9C97418000FD for ; Tue, 13 Feb 2024 17:48:59 -0500 (EST) Received: from imap49 ([10.202.2.99]) by compute3.internal (MEProxy); Tue, 13 Feb 2024 17:48:59 -0500 DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d= messagingengine.com; h=cc:content-type:content-type:date:date :feedback-id:feedback-id:from:from:in-reply-to:in-reply-to :message-id:mime-version:references:reply-to:subject:subject:to :to:x-me-proxy:x-me-proxy:x-me-sender:x-me-sender:x-sasl-enc; s= fm3; t=1707864539; x=1707950939; bh=9ayQJvj8Rqv8KE/pl4bh6xCHEFOs CAgpD2bg5UbdRkY=; b=HWNZrkTMDnO6HBRbZ6xxmyAw3JCAh1IutN3RohGdGrFe mwJqoSpVOQk847UZ0UAvg2s70wmS57vsmbDkEpWsHOAyNcoKo4Avz0Ycaw4LgZxr oa5n3AMlCJOieiwWDhBRrQXMaGDX+Y3CTakqnChXUYbBpQp9qPcSs9QmeSbrK3Fs 7UItXPGRoMn66Omu8IV/gXkKg7j18D242kwKY4XKui0B/y3M9Yp6k+suGutu5RNA MBP1oh+2JPtuhub4YSHLKdOIzVUFpKCa1WsmkXMxluh6unKgCUldNRgFNeX4BHL3 dzBKU0JSpQjsiX2AGtw59G7/wn8JLRzq7Vr8qk87gw== X-ME-Sender: X-ME-Proxy-Cause: gggruggvucftvghtrhhoucdtuddrgedvledrudeigddtgecutefuodetggdotefrodftvf curfhrohhfihhlvgemucfhrghsthforghilhdpqfgfvfdpuffrtefokffrpgfnqfghnecu uegrihhlohhuthemuceftddtnecunecujfgurhepofgfggfkjghffffhvffutgesthdtre dtreertdenucfhrhhomhepfdflvggrnhdquegrphhtihhsthgvucfmvghmphhffdcuoehj sgesvhhiuggvohhlrghnrdhorhhgqeenucggtffrrghtthgvrhhnpeeiiefgvdffteekke ffvdeiffeijeethfevteeutdegvdegfeekgfektdelheefgfenucevlhhushhtvghrufhi iigvpedtnecurfgrrhgrmhepmhgrihhlfhhrohhmpehjsgesvhhiuggvohhlrghnrdhorh hg X-ME-Proxy: Feedback-ID: i06904239:Fastmail Received: by mailuser.nyi.internal (Postfix, from userid 501) id ACA9415A0092; Tue, 13 Feb 2024 17:48:58 -0500 (EST) X-Mailer: MessagingEngine.com Webmail Interface User-Agent: Cyrus-JMAP/3.11.0-alpha0-144-ge5821d614e-fm-20240125.002-ge5821d61 MIME-Version: 1.0 Message-Id: In-Reply-To: <20240202072629.1259512-1-wenbin.chen@intel.com> References: <20240202072629.1259512-1-wenbin.chen@intel.com> Date: Tue, 13 Feb 2024 23:48:38 +0100 From: "Jean-Baptiste Kempf" To: ffmpeg-devel Subject: Re: [FFmpeg-devel] [PATCH v2] libavfi/dnn: add LibTorch as one of DNN backend X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Content-Type: text/plain; charset="us-ascii" Content-Transfer-Encoding: 7bit Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Archived-At: List-Archive: List-Post: Hello, On Fri, 2 Feb 2024, at 08:26, wenbin.chen-at-intel.com@ffmpeg.org wrote: > +static void infer_completion_callback(void *args) { > + THRequestItem *request = (THRequestItem*)args; > + LastLevelTaskItem *lltask = request->lltask; > + TaskItem *task = lltask->task; > + DNNData outputs = { 0 }; > + THInferRequest *infer_request = request->infer_request; > + THModel *th_model = (THModel *)task->model; > + torch::Tensor *output = infer_request->output; > + > + c10::IntArrayRef sizes = output->sizes(); > + assert(sizes.size == 5); Why 5? > + outputs.order = DCO_RGB; > + outputs.layout = DL_NCHW; > + outputs.dims[2] = sizes.at(3); > + outputs.dims[3] = sizes.at(4); > + outputs.dt = DNN_FLOAT; > + outputs.dims[1] = 3; Why 3? -- Jean-Baptiste Kempf - President +33 672 704 734 _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".