Searched refs:m_inputStrides (Results 1 - 8 of 8) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorStriding.h130 m_inputStrides[0] = 1;
133 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];
134 m_inputStrides[i-1] *= op.strides()[i-1];
136 m_inputStrides[NumDims-1] *= op.strides()[NumDims-1];
139 m_inputStrides[NumDims-1] = 1;
142 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];
143 m_inputStrides[i+1] *= op.strides()[i+1];
145 m_inputStrides[
251 array<Index, NumDims> m_inputStrides; member in struct:Eigen::TensorEvaluator
[all...]
H A DTensorBroadcasting.h130 m_inputStrides[0] = 1;
133 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];
137 m_inputStrides[NumDims-1] = 1;
140 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];
178 inputIndex += idx * m_inputStrides[i];
183 inputIndex += (idx % m_impl.dimensions()[i]) * m_inputStrides[i];
208 inputIndex += idx * m_inputStrides[i];
213 inputIndex += (idx % m_impl.dimensions()[i]) * m_inputStrides[
[all...]
H A DTensorPatch.h114 m_inputStrides[0] = 1;
117 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];
131 m_inputStrides[NumDims-2] = 1;
134 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];
169 inputIndex += (patchIdx + offsetIdx) * m_inputStrides[i];
177 inputIndex += (patchIdx + offsetIdx) * m_inputStrides[i];
210 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];
211 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[
261 array<Index, NumDims-1> m_inputStrides; member in struct:Eigen::TensorEvaluator
[all...]
H A DTensorInflation.h115 m_inputStrides[0] = 1;
118 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];
122 m_inputStrides[NumDims-1] = 1;
125 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];
152 *inputIndex += idx / m_strides[i] * m_inputStrides[i];
166 *inputIndex += idx / m_strides[i] * m_inputStrides[i];
221 array<Index, NumDims> m_inputStrides; member in namespace:Eigen
H A DTensorMorphing.h331 m_inputStrides[0] = 1;
333 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];
343 m_inputStrides[NumDims-1] = 1;
345 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];
421 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
422 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];
432 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];
433 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[
521 array<Index, NumDims> m_inputStrides; member in struct:Eigen::TensorEvaluator
840 array<Index, NumDims> m_inputStrides; member in struct:Eigen::TensorEvaluator
[all...]
H A DTensorShuffling.h147 m_inputStrides[i] = inputStrides[shuffle[i]];
196 inputIndex += idx * m_inputStrides[i];
199 return inputIndex + index * m_inputStrides[0];
203 inputIndex += idx * m_inputStrides[i];
206 return inputIndex + index * m_inputStrides[NumDims - 1];
212 array<Index, NumDims> m_inputStrides; member in struct:Eigen::TensorEvaluator
H A DTensorPadding.h118 m_inputStrides[0] = 1;
121 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];
126 m_inputStrides[NumDims - 1] = 1;
129 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];
156 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];
169 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];
283 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];
341 inputIndex += (idx - m_padding[i].first) * m_inputStrides[
385 array<Index, NumDims> m_inputStrides; member in namespace:Eigen
[all...]
H A DTensorConvolution.h85 m_inputStrides[i] = inputStrides[ordering[i]];
121 inputIndex += idx * m_inputStrides[d];
124 inputIndex += p * m_inputStrides[NumKernelDims];
132 inputIndex += idx * m_inputStrides[d];
135 inputIndex += p * m_inputStrides[limit];
168 return i * m_inputStrides[offset];
182 return i * m_inputStrides[offset] + j * m_inputStrides[offset + 1];
196 return i * m_inputStrides[offset] + j * m_inputStrides[offse
210 array<Index, NumDims> m_inputStrides; member in class:Eigen::internal::IndexMapper
[all...]

Completed in 556 milliseconds