Lines Matching full:ndim
21 auto ndim = dimsA > dimsB ? dimsA : dimsB; in infer_size_impl() local
22 Container expandedSizes(ndim); in infer_size_impl()
24 for (ptrdiff_t i = ndim - 1; i >= 0; --i) { in infer_size_impl()
25 ptrdiff_t offset = ndim - 1 - i; in infer_size_impl()
66 int64_t ndim = static_cast<int64_t>(sizes.size()); in inferExpandGeometryImpl() local
70 return InferExpandGeometryResult<Container>(sizes, ndim); in inferExpandGeometryImpl()
73 InferExpandGeometryResult<Container> result(ndim); in inferExpandGeometryImpl()
78 for (int64_t i = ndim - 1; i >= 0; --i) { in inferExpandGeometryImpl()
79 int64_t offset = ndim - 1 - i; in inferExpandGeometryImpl()
153 size_t ndim = tensor_sizes.size(); in infer_dense_strides() local
154 if (ndim == 0) { in infer_dense_strides()
157 if (ndim == 1) { in infer_dense_strides()
161 std::vector<int64_t> perm(ndim); in infer_dense_strides()
197 for (const auto i : c10::irange(1, ndim)) { in infer_dense_strides()
213 std::vector<int64_t> out_strides(ndim); in infer_dense_strides()
215 for (const auto i : c10::irange(ndim)) { in infer_dense_strides()