Home
last modified time | relevance | path

Searched refs:nested_sizes2 (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.cpp985 static bool can_cat_nested_sizes(const Tensor& nested_sizes1, const Tensor& nested_sizes2, int64_t … in can_cat_nested_sizes() argument
986 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()