search

rocmarchive / realcaffe2

The repo is obsolete. Use at your own risk.
https://github.com/pytorch/pytorch
Apache License 2.0
12 stars 2 forks source link

share the workspace between fwd/bwd pass for pooling op #84

Closed petrex closed 6 years ago

petrex commented 6 years ago

@whchung Let me talk to you offline to get your insight here.

whchung commented 6 years ago

@petrex check how i did backward pooling in TensorFlow: https://github.com/ROCmSoftwarePlatform/tensorflow/blob/rocm-v1/tensorflow/stream_executor/rocm/rocm_dnn.cc#L2782

it's not elegant but it works

petrex commented 6 years ago

97