Searched refs:_stack_if_compiling (Results 1 – 4 of 4) sorted by relevance
19 _stack_if_compiling,554 step_size_grads = _stack_if_compiling(560 step_size_expavg = _stack_if_compiling(
20 _stack_if_compiling,591 step_size = _stack_if_compiling([(lr / bc) * -1 for bc in bias_correction1])
20 _stack_if_compiling,599 step_size = _stack_if_compiling([(lr / bc) * -1 for bc in bias_correction1])
109 def _stack_if_compiling(x): function