Home
last modified time | relevance | path

Searched refs:_stack_if_compiling (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/optim/
H A Dnadam.py19 _stack_if_compiling,
554 step_size_grads = _stack_if_compiling(
560 step_size_expavg = _stack_if_compiling(
H A Dadamw.py20 _stack_if_compiling,
591 step_size = _stack_if_compiling([(lr / bc) * -1 for bc in bias_correction1])
H A Dadam.py20 _stack_if_compiling,
599 step_size = _stack_if_compiling([(lr / bc) * -1 for bc in bias_correction1])
H A Doptimizer.py109 def _stack_if_compiling(x): function