Lines Matching refs:dims
25 struct Dims {
31 inline bool NextIndex(const int num_dims, const int* dims, int* current) {
33 TFLITE_DCHECK(dims != nullptr);
38 TFLITE_DCHECK_GE(dims[idx], current_val);
39 if (dims[idx] == current_val) {
55 // TODO(kanlig): uses Dims to represent dimensions.
56 inline size_t ReducedOutputOffset(const int num_dims, const int* dims,
60 TFLITE_DCHECK(dims != nullptr);
75 offset = offset * static_cast<size_t>(dims[idx]) +
82 inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3) {
83 TFLITE_DCHECK(i0 >= 0 && i0 < dims.sizes[0]);
84 TFLITE_DCHECK(i1 >= 0 && i1 < dims.sizes[1]);
85 TFLITE_DCHECK(i2 >= 0 && i2 < dims.sizes[2]);
86 TFLITE_DCHECK(i3 >= 0 && i3 < dims.sizes[3]);
87 return i0 * dims.strides[0] + i1 * dims.strides[1] + i2 * dims.strides[2] +
88 i3 * dims.strides[3];
91 inline int Offset(const Dims<4>& dims, int* index) {
92 return Offset(dims, index[0], index[1], index[2], index[3]);
97 int ArraySize(const Dims<N>& array, int index) {
117 inline int RequiredBufferSizeForDims(const Dims<4>& dims) {
120 max_offset += (dims.sizes[i] - 1) * dims.strides[i];
126 bool IsPackedWithoutStrides(const Dims<N>& dims) {
129 if (dims.strides[d] != expected_stride) return false;
130 expected_stride *= dims.sizes[d];
