Searched refs:get_consistent_last_dim_of_nested_tensor (Results 1 – 3 of 3) sorted by relevance
60 int64_t get_consistent_last_dim_of_nested_tensor(const NestedTensorImpl& nt) { in get_consistent_last_dim_of_nested_tensor() function80 const int64_t last_dim_size = get_consistent_last_dim_of_nested_tensor(*self_impl); in chunk_nested_tensor()135 auto self_size = get_consistent_last_dim_of_nested_tensor(*self_impl); in split_with_sizes_nested()
38 const auto last_dim = get_consistent_last_dim_of_nested_tensor(*nt_input); in check_nested_tensor_matrix_constraints()104 get_consistent_last_dim_of_nested_tensor(*nt_mat1) == mat2.sizes()[0]); in NestedTensor_times_Tensor_plus_Tensor_addmm()
105 int64_t get_consistent_last_dim_of_nested_tensor(const NestedTensorImpl& nt);