Searched defs:nt_grad (Results 1 – 1 of 1) sorted by relevance
124 auto nt_grad = get_nested_tensor_impl(grad); in _nested_sum_backward_cpu() local173 auto nt_grad = wrap_buffer(self_grad_buffer, self_sizes); in _nested_select_backward_symint() local