Searched refs:nested_tensor_impl_is_contiguous (Results 1 – 6 of 6) sorted by relevance
28 nested_tensor_impl_is_contiguous(nt_input), in check_nested_tensor_matrix_constraints()101 TORCH_INTERNAL_ASSERT(nested_tensor_impl_is_contiguous(nt_mat1)); in NestedTensor_times_Tensor_plus_Tensor_addmm()136 nested_tensor_impl_is_contiguous(&nt_self) && in NestedTensor_add_NestedTensor_in_place()137 nested_tensor_impl_is_contiguous(&nt_other)); in NestedTensor_add_NestedTensor_in_place()145 TORCH_INTERNAL_ASSERT(nested_tensor_impl_is_contiguous(query_nt)); in NestedTensor_softmax_dropout()215 …TORCH_CHECK(nested_tensor_impl_is_contiguous(nt_impl), "to_mask only works on contiguous NestedTen… in NestedTensor_to_mask()
159 TORCH_CHECK(nested_tensor_impl_is_contiguous(nt_input)); in nested_layer_norm()252 nested_tensor_impl_is_contiguous(get_nested_tensor_impl(t)), in NestedTensor_to_padded_tensor_generic()340 TORCH_CHECK(nested_tensor_impl_is_contiguous(nt_indices)); in NestedTensor_embedding()383 nested_tensor_impl_is_contiguous(nt_input), in NestedTensor_sum_dim_CPU()
53 TORCH_INTERNAL_ASSERT(nested_tensor_impl_is_contiguous(nt_grad_output)); in nested_linear_backward()
15 inline bool nested_tensor_impl_is_contiguous(const NestedTensorImpl* nt);85 nested_tensor_impl_is_contiguous(this), in get_buffer()224 inline bool nested_tensor_impl_is_contiguous(const NestedTensorImpl* nt) { in nested_tensor_impl_is_contiguous() function
275 return nested_tensor_impl_is_contiguous(this); in is_contiguous_custom()
150 nested_tensor_impl_is_contiguous(nt_input), in NestedTensor_to_padded_tensor_cuda()