pytorch/caffe2/operators/hip
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
..
activation_ops_miopen.h [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -07:00
conv_op_miopen.hip [ROCm] Enable 3D convolutions through ROCm (#33067) 2020-02-14 13:19:10 -08:00
conv_transpose_op_miopen.hip
elu_op_miopen.hip
local_response_normalization_op_miopen.hip
pool_op_miopen.hip [ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260) 2020-07-08 17:42:55 -07:00
relu_op_miopen.hip
sigmoid_op_miopen.hip
spatial_batch_norm_op_miopen.hip [ROCm] Enable 3D batch norms through MIOpen (#33262) 2020-02-13 11:29:51 -08:00
tanh_op_miopen.hip