@@ -150,7 +150,7 @@ TEST(VideoDecoderTest, RespectsWidthAndHeightFromOptions) {
150150 videoStreamOptions.width = 100 ;
151151 videoStreamOptions.height = 120 ;
152152 decoder->addVideoStreamDecoder (-1 , videoStreamOptions);
153- torch::Tensor tensor = decoder->getNextFrameNoDemux ().frame ;
153+ torch::Tensor tensor = decoder->getNextFrameNoDemux ().data ;
154154 EXPECT_EQ (tensor.sizes (), std::vector<long >({3 , 120 , 100 }));
155155}
156156
@@ -161,7 +161,7 @@ TEST(VideoDecoderTest, RespectsOutputTensorDimensionOrderFromOptions) {
161161 VideoDecoder::VideoStreamOptions videoStreamOptions;
162162 videoStreamOptions.dimensionOrder = " NHWC" ;
163163 decoder->addVideoStreamDecoder (-1 , videoStreamOptions);
164- torch::Tensor tensor = decoder->getNextFrameNoDemux ().frame ;
164+ torch::Tensor tensor = decoder->getNextFrameNoDemux ().data ;
165165 EXPECT_EQ (tensor.sizes (), std::vector<long >({270 , 480 , 3 }));
166166}
167167
@@ -235,7 +235,7 @@ TEST_P(VideoDecoderTest, DecodesFramesInABatchInNHWC) {
235235 VideoDecoder::VideoStreamOptions (" dimension_order=NHWC" ));
236236 // Frame with index 180 corresponds to timestamp 6.006.
237237 auto output = ourDecoder->getFramesAtIndices (bestVideoStreamIndex, {0 , 180 });
238- auto tensor = output.;
238+ auto tensor = output.data ;
239239 EXPECT_EQ (tensor.sizes (), std::vector<long >({2 , 270 , 480 , 3 }));
240240
241241 torch::Tensor tensor0FromFFMPEG =
0 commit comments