Searched refs:nested_sizes1 (Results 1 – 1 of 1) sorted by relevance
985 static bool can_cat_nested_sizes(const Tensor& nested_sizes1, const Tensor& nested_sizes2, int64_t … in can_cat_nested_sizes() argument986 if (nested_sizes1.sizes() != nested_sizes2.sizes()) { in can_cat_nested_sizes()990 auto nested_sizes1_ptr = nested_sizes1.data_ptr<int64_t>(); in can_cat_nested_sizes()992 const auto num_components = nested_sizes1.size(0); in can_cat_nested_sizes()993 const auto num_dims = nested_sizes1.size(1); in can_cat_nested_sizes()