Searched defs:accu_grad (Results 1 – 2 of 2) sorted by relevance
438 def tensor_grad_scale_pipeline(scale, grad, accu_grad): argument448 def tensor_shard_grad_scale_pipeline(scale, grad, accu_grad): argument
494 def _pipeline_clear_grad(accu_grad, grad): argument