Searched defs:num_reduce_dims (Results 1 – 2 of 2) sorted by relevance
228 int num_reduce_dims = iter.num_reduce_dims(); in make_output_calculator() local242 int num_reduce_dims = iter.num_reduce_dims(); in make_input_calculator() local
752 int TensorIteratorBase::num_reduce_dims() const { in num_reduce_dims() function in at::TensorIteratorBase