Searched defs:stop_gradients (Results 1 – 3 of 3) sorted by relevance
47 stop_gradients=None, argument179 stop_gradients=None, argument
475 stop_gradients=None, argument
352 def _MakeGraph(rng, stop_gradients=()): argument