Lines Matching full:numel
126 void checkNumel(CheckedFrom c, const TensorGeometryArg& t, int64_t numel) { in checkNumel() argument
128 t->numel() == numel, in checkNumel()
129 "Expected tensor for ", t, " to have ", numel, in checkNumel()
130 " elements; but it actually has ", t->numel(), " elements", in checkNumel()
136 t1->numel() == t2->numel(), in checkSameNumel()
139 t1->numel(), " does not equal ", t2->numel(), in checkSameNumel()
323 // ``numel'', i.e., number of subspaces, as the corresponding chunk of
329 template <typename ResultVec, typename NewShapeVec, typename Numel>
340 // NOTE: stride is arbitrary in the numel() == 0 case; in computeStride_impl()
345 const Numel numel = c10::multiply_integers(oldshape); in computeStride_impl() local
346 bool zero_numel = TORCH_GUARD_SIZE_OBLIVIOUS(sym_eq(numel, 0)); in computeStride_impl()
358 std::max<Numel>(newshape[view_d+1], Numel(1)) * newstride[view_d+1]; in computeStride_impl()
366 Numel chunk_base_stride = oldstride.back(); in computeStride_impl()
367 // numel in current chunk in computeStride_impl()
368 Numel tensor_numel = 1; in computeStride_impl()
369 Numel view_numel = 1; in computeStride_impl()