Searched refs:grad_input_buffer (Results 1 – 1 of 1) sorted by relevance
60 auto grad_input_buffer = at::mm(reshaped_grad, weight).view({-1}); in nested_linear_backward() local62 grad_input = wrap_buffer(grad_input_buffer, grad_input_nt_size); in nested_linear_backward()