Searched refs:get_dim_to_split (Results 1 – 2 of 2) sorted by relevance
1326 int TensorIteratorBase::get_dim_to_split() const { in get_dim_to_split() function in at::TensorIteratorBase1684 auto split_dim = iter.get_dim_to_split(); in operator ++()
381 int get_dim_to_split() const;