Searched refs:nested_sizes2 (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()991 auto nested_sizes2_ptr = nested_sizes2.data_ptr<int64_t>(); in can_cat_nested_sizes()