Improve usability of `tf.contrib.bayesflow.custom_gradient` by removing need for...
authorJoshua V. Dillon <jvdillon@google.com>
Mon, 12 Mar 2018 18:29:24 +0000 (11:29 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 12 Mar 2018 18:33:33 +0000 (11:33 -0700)
commit617d1f01d60b677536f988be35dc4f02885e6f1e
tree0cf0d88465195137e3f96650bdb86ba037a9f8f9
parent402fb8c97db05b51587c6fc999c690d548fd4496
Improve usability of `tf.contrib.bayesflow.custom_gradient` by removing need for `axis` arg and support taking lists.

PiperOrigin-RevId: 188751894
tensorflow/contrib/bayesflow/python/kernel_tests/custom_grad_test.py
tensorflow/contrib/bayesflow/python/ops/custom_grad_impl.py