mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
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 |
||
|---|---|---|
| .. | ||
| activation_ops_miopen.h | ||
| conv_op_miopen.hip | ||
| conv_transpose_op_miopen.hip | ||
| elu_op_miopen.hip | ||
| local_response_normalization_op_miopen.hip | ||
| pool_op_miopen.hip | ||
| relu_op_miopen.hip | ||
| sigmoid_op_miopen.hip | ||
| spatial_batch_norm_op_miopen.hip | ||
| tanh_op_miopen.hip | ||