Searched refs:nt_impl_input (Results 1 – 1 of 1) sorted by relevance
207 auto* nt_impl_input = get_nested_tensor_impl(input); in layer_norm_backward_nested() local210 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()