Searched defs:AccumulateGrad (Results 1 – 2 of 2) sorted by relevance
40 struct TORCH_API AccumulateGrad : public Node { struct47 // NB: Since the AccumulateGrad Node is only a weak ref from the Tensor, in tensor_pre_hooks() argument53 std::unique_ptr<PostAccumulateGradHook>& tensor_post_acc_grad_hooks() noexcept in tensor_post_acc_grad_hooks() argument115 static void accumulateGrad( in accumulateGrad()270 Variable variable;
19 AccumulateGrad::AccumulateGrad(Variable variable_) in AccumulateGrad() function in torch::autograd::AccumulateGrad