Searched defs:stop_gradients (Results 1 – 3 of 3) sorted by relevance
51 stop_gradients=None, argument183 stop_gradients=None, argument
487 stop_gradients=None, argument
353 def _MakeGraph(rng, stop_gradients=()): argument