Lines Matching refs:m_offsets
322 : m_impl(op.expression(), device), m_device(device), m_dimensions(op.sizes()), m_offsets(op.startIndices())
421 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
422 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
426 inputIndices[0] += (indices[0] + m_offsets[0]);
427 inputIndices[1] += (indices[1] + m_offsets[0]);
432 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
433 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
437 m_offsets[NumDims-1]);
438 inputIndices[1] += (indices[1] + m_offsets[NumDims-1]);
468 offset += m_offsets[i] * m_inputStrides[i];
473 offset += m_offsets[j] * m_inputStrides[j];
481 offset += m_offsets[i] * m_inputStrides[i];
486 offset += m_offsets[j] * m_inputStrides[j];
504 inputIndex += (idx + m_offsets[i]) * m_inputStrides[i];
507 inputIndex += (index + m_offsets[0]);
511 inputIndex += (idx + m_offsets[i]) * m_inputStrides[i];
514 inputIndex += (index + m_offsets[NumDims-1]);
525 const StartIndices m_offsets;
571 inputIndices[0] += (idx0 + this->m_offsets[i]) * this->m_inputStrides[i];
572 inputIndices[1] += (idx1 + this->m_offsets[i]) * this->m_inputStrides[i];
576 inputIndices[0] += (indices[0] + this->m_offsets[0]);
577 inputIndices[1] += (indices[1] + this->m_offsets[0]);
582 inputIndices[0] += (idx0 + this->m_offsets[i]) * this->m_inputStrides[i];
583 inputIndices[1] += (idx1 + this->m_offsets[i]) * this->m_inputStrides[i];
587 inputIndices[0] += (indices[0] + this->m_offsets[NumDims-1]);
588 inputIndices[1] += (indices[1] + this->m_offsets[NumDims-1]);
745 m_offsets[0] = startIndicesClamped[0];
750 m_offsets[i] = startIndicesClamped[i] * previousDimProduct;
762 m_offsets[NumDims-1] = startIndicesClamped[NumDims-1];
767 m_offsets[i] = startIndicesClamped[i] * previousDimProduct;
821 inputIndex += idx * m_inputStrides[i] + m_offsets[i];
827 inputIndex += idx * m_inputStrides[i] + m_offsets[i];
845 DSizes<Index, NumDims> m_offsets; // offset in a flattened shape
