Searched defs:grad_distributed (Results 1 – 1 of 1) sorted by relevance
6881 Tensor grad_distributed = grad / N_to_distribute; in scatter_reduce_backward() local6976 Tensor grad_distributed = grad / N_to_distribute; in index_reduce_backward() local