Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/nested/
DNestedTensorBackward.cpp207 auto* nt_impl_input = get_nested_tensor_impl(input); in layer_norm_backward_nested() local
210 const auto& sizes = nt_impl_input->get_nested_sizes(); in layer_norm_backward_nested()
212 *nt_impl_input, normalized_shape, weight, bias); in layer_norm_backward_nested()
222 auto input_buffer = nt_impl_input->get_buffer(); in layer_norm_backward_nested()