Searched defs:nt_input (Results 1 – 5 of 5) sorted by relevance
19 auto* nt_input = get_nested_tensor_impl(nested_tensor); in check_nested_tensor_matrix_constraints() local63 auto* nt_input = get_nested_tensor_impl(input); in nested_linear() local
171 auto* nt_input = get_nested_tensor_impl(self); in _pin_memory_nested() local
158 auto* nt_input = get_nested_tensor_impl(input); in nested_layer_norm() local381 auto nt_input = get_nested_tensor_impl(self); in NestedTensor_sum_dim_CPU() local
50 auto* nt_input = get_nested_tensor_impl(input); in nested_linear_backward() local
148 auto* nt_input = get_nested_tensor_impl(t); in NestedTensor_to_padded_tensor_cuda() local