OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
as_tensor_docs
pytorch/caffe2/video
History
lcskrishna 302cf6835e [ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260) 
Summary:
This PR contains the following updates:
1. MIOpen 3D pooling enabled in Caffe2.
2. Refactored the MIOpen pooling code in caffe2.
3. Enabled unit test cases for 3D pooling.

CC: ezyang jeffdaily ashishfarmer

Pull Request resolved: https://github.com/pytorch/pytorch/pull/38260

Differential Revision: D21524754

Pulled By: xw285cornell

fbshipit-source-id: ddfe09dc585cd61e42eee22eff8348d326fd0c3b
2020-07-08 17:42:55 -07:00
..
CMakeLists.txt [ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260) 2020-07-08 17:42:55 -07:00
optical_flow.cc
optical_flow.h
video_decoder.cc Fix compilation error when buildng with FFMPEG (#27589) 2020-02-13 11:23:48 -08:00
video_decoder.h Add CAFFE2_API to video decoding functions (#31187) 2020-01-15 19:39:02 -08:00
video_input_op_gpu.cc
video_input_op.cc update video input (#22471) 2019-07-05 00:56:33 -07:00
video_input_op.h Fix compilation error when buildng with FFMPEG (#27589) 2020-02-13 11:23:48 -08:00
video_io.cc update video input (#22471) 2019-07-05 00:56:33 -07:00
video_io.h update video input (#22471) 2019-07-05 00:56:33 -07:00