Home
last modified time | relevance | path

Searched defs:self_grad_buffer (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorBackward.cpp130 const Tensor& self_grad_buffer = self_buffer.new_empty(self_buffer.sizes()); in _nested_sum_backward_cpu() local
171 const Tensor& self_grad_buffer = self_buffer.new_zeros(self_buffer.sizes()); in _nested_select_backward_symint() local