C++ gradients: Fractional*Pool, Soft{Plus,Sign}
authorKB Sriram <kbsriram@gmail.com>
Wed, 28 Feb 2018 15:16:20 +0000 (07:16 -0800)
committerKB Sriram <kbsriram@gmail.com>
Wed, 28 Feb 2018 15:22:34 +0000 (07:22 -0800)
commit75259500f80266998f232a94853b0bc08d2925cc
tree543550aa683f1f25dedf1640ace2a670fade5409
parent3ba1f72f8829c566372208062fcea04ab5695dc6
C++ gradients: Fractional*Pool, Soft{Plus,Sign}

1. Adds gradients for four nn ops:
   FractionalAvgPool
   FractionalMaxPool
   SoftPlus
   SoftSign

2. Update randomization to allow numeric gradient
   checks on max pooling algorithms with more than
   one pool.

Resolves https://github.com/tensorflow/tensorflow/issues/17330
tensorflow/cc/gradients/nn_grad.cc
tensorflow/cc/gradients/nn_grad_test.cc