Searched defs:grad_buffer (Results 1 – 1 of 1) sorted by relevance
125 const Tensor& grad_buffer = nt_grad->get_buffer(); in _nested_sum_backward_cpu() local223 auto grad_buffer = nt_impl_grad->get_buffer(); in layer_norm_backward_nested() local