Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp1326 int TensorIteratorBase::get_dim_to_split() const { in get_dim_to_split() function in at::TensorIteratorBase
1684 auto split_dim = iter.get_dim_to_split(); in operator ++()
H A DTensorIterator.h381 int get_dim_to_split() const;