List-of-Differentiable--OPs-and-Non-differentiable-OPs--in-Tensorflow

List-of-Differentiable--OPs-and-Non-differentiable-OPs--in-Tensorflow copied to clipboard

Round is a differentiable operation?

1

I thought there was no gradient defined in TensorFlow for round?

scottzockoll

I think currently the list of differentiable OPs is incorrect. Take `ArgMax` as an example: https://github.com/tensorflow/tensorflow/blob/87462bfac761435a46641ff2f10ad0b6e5414a4b/tensorflow/python/ops/math_grad.py#L37 It returns [None, None]. And I guess, that is why I'm still getting: `ValueError:...

TheCrazyT

Note for python tensorflow API users... to find the tensorflow equivalent class names, you can search for them in tensorflow/python/ops/array_ops.py, and look for which tensorflow operator it is getting mapped...

drewm1980

Alternatively, you could just focus on the list of known differentiable functions, and include a "since" flag when you detect a version that doesn't have AD. Your script ran successfully...

drewm1980