Lines Matching defs:subs
197 // Collect dimension-wise indices (subs).
198 array<Index, NumDims> subs;
201 subs[i] = index / m_outputStrides[i];
202 index -= subs[i] * m_outputStrides[i];
204 subs[0] = index;
207 subs[i] = index / m_outputStrides[i];
208 index -= subs[i] * m_outputStrides[i];
210 subs[NumDims - 1] = index;
214 if (subs[m_axis] < left_dims[m_axis]) {
217 left_index = subs[0];
219 left_index += (subs[i] % left_dims[i]) * m_leftStrides[i];
222 left_index = subs[NumDims - 1];
224 left_index += (subs[i] % left_dims[i]) * m_leftStrides[i];
229 subs[m_axis] -= left_dims[m_axis];
233 right_index = subs[0];
235 right_index += (subs[i] % right_dims[i]) * m_rightStrides[i];
238 right_index = subs[NumDims - 1];
240 right_index += (subs[i] % right_dims[i]) * m_rightStrides[i];
318 // Collect dimension-wise indices (subs).
319 array<Index, Base::NumDims> subs;
321 subs[i] = index / this->m_outputStrides[i];
322 index -= subs[i] * this->m_outputStrides[i];
324 subs[0] = index;
327 if (subs[this->m_axis] < left_dims[this->m_axis]) {
328 Index left_index = subs[0];
330 left_index += (subs[i] % left_dims[i]) * this->m_leftStrides[i];
334 subs[this->m_axis] -= left_dims[this->m_axis];
336 Index right_index = subs[0];
338 right_index += (subs[i] % right_dims[i]) * this->m_rightStrides[i];
