Consumers() in TensorFlow.jl

Specific Domains Machine Learning
1

I would like to implement this method in TensorFlow.jl
https://github.com/ProbabilisticNumerics/cabs
which should automatically adjust the size of the minibatch in SGD. I am curious about the method, since in security the signals are typically very weak and therefore I would expect that large batch sizes would be needed. Looking at the source code
https://github.com/ProbabilisticNumerics/cabs/blob/master/gradient_moment.py
it seems like they need to identify consumers of a tensor (line 79 of gradient_moment.py).
I wonder if it is possible to do so in?

Thanks for help.
Tomas