Searched refs:inputStrides (Results 1 – 2 of 2) sorted by relevance
128 array<Index, NumDims> inputStrides;131 inputStrides[0] = 1;134 inputStrides[i] = inputStrides[i - 1] * input_dims[i - 1];138 inputStrides[NumDims - 1] = 1;141 inputStrides[i] = inputStrides[i + 1] * input_dims[i + 1];147 m_inputStrides[i] = inputStrides[shuffle[i]];
39 array<Index, NumDims> inputStrides; in IndexMapper() local42 inputStrides[0] = 1; in IndexMapper()45 inputStrides[i] = inputStrides[i-1] * input_dims[i-1]; in IndexMapper()49 inputStrides[NumDims - 1] = 1; in IndexMapper()52 inputStrides[i] = inputStrides[i + 1] * input_dims[i + 1]; in IndexMapper()85 m_inputStrides[i] = inputStrides[ordering[i]]; in IndexMapper()