Home
last modified time | relevance | path

Searched refs:nested_self (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/nested/
DNestedTensorBackward.cpp120 const Tensor& nested_self, in _nested_sum_backward_cpu() argument
123 auto nt_self = get_nested_tensor_impl(nested_self); in _nested_sum_backward_cpu()
164 const Tensor& nested_self, in _nested_select_backward_symint() argument
168 auto nt_self = get_nested_tensor_impl(nested_self); in _nested_select_backward_symint()