Searched defs:new_grad (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/utils/ |
| H A D | grad_layout_contract.h | 58 const at::Tensor& new_grad, in clone_obey_contract()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/ |
| H A D | accumulate_grad.cpp | 36 at::Tensor new_grad = std::move(grads[0]); in apply() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd_meta.cpp | 187 Tensor new_grad(new_grad_base); in set_fw_grad() local
|
| H A D | FunctionsManual.cpp | 6089 auto new_grad = at::where(mask_first_zero, self_t, self_p).cumprod(dim); in cumprod_jvp() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/ |
| H A D | inductor_ops.cpp | 72 static void accumulate_grad_(const Tensor& variable, const Tensor& new_grad) { in accumulate_grad_()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/polyfills/ |
| H A D | __init__.py | 50 def accumulate_grad(x, new_grad): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/templates/ |
| H A D | TensorBody.h | 511 void _set_fw_grad(const TensorBase& new_grad, uint64_t level, bool is_inplace_op) const { in _set_fw_grad()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TensorBase.h | 791 void _set_fw_grad(const TensorBase& new_grad, uint64_t level, bool is_inplace_op) const { in _set_fw_grad()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | TensorImpl.cpp | 68 const at::TensorBase& new_grad, in _set_fw_grad()
|