Searched refs:dim_start (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/ |
| D | TensorUtils.cpp | 48 void checkDimRange(CheckedFrom c, const TensorGeometryArg& t, int64_t dim_start, int64_t dim_end) { in checkDimRange() argument
50 t->dim() >= dim_start && t->dim() < dim_end, in checkDimRange()
51 "Expected ", dim_start, " to ", (dim_end - 1), " dimensions, but got ", in checkDimRange()
|
| D | TensorUtils.h | 85 int64_t dim_start,
|