Searched defs:grad_input_acc (Results 1 – 2 of 2) sorted by relevance
360 auto grad_input_acc = grad_input.accessor<scalar_t, 2>(); in nll_loss_backward_out_frame() local381 auto grad_input_acc = grad_input.accessor<scalar_t, 1>(); in nll_loss_backward_out_frame() local392 auto grad_input_acc = grad_input.accessor<scalar_t, 2>(); in nll_loss_backward_out_frame() local
302 auto grad_input_acc = grad_input.accessor<scalar_t, 4>(); in nll_loss2d_backward_out_frame() local