HomeSort by relevance Sort by last modified time
    Searched refs:m_outputStrides (Results 1 - 10 of 10) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorStriding.h 129 m_outputStrides[0] = 1;
132 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
138 m_outputStrides[NumDims-1] = 1;
141 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
174 const Index idx0 = indices[0] / m_outputStrides[i];
175 const Index idx1 = indices[1] / m_outputStrides[i];
178 indices[0] -= idx0 * m_outputStrides[i];
179 indices[1] -= idx1 * m_outputStrides[i]
    [all...]
TensorPatch.h 120 m_outputStrides[0] = 1;
122 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
137 m_outputStrides[NumDims-1] = 1;
139 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
159 Index patchIndex = index / m_outputStrides[output_stride_index];
161 Index patchOffset = index - patchIndex * m_outputStrides[output_stride_index];
167 const Index offsetIdx = patchOffset / m_outputStrides[i];
168 patchOffset -= offsetIdx * m_outputStrides[i]
    [all...]
TensorPadding.h 119 m_outputStrides[0] = 1;
122 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
124 m_outputStrides[NumDims] = m_outputStrides[NumDims-1] * m_dimensions[NumDims-1];
127 m_outputStrides[NumDims] = 1;
130 m_outputStrides[i+1] = m_outputStrides[i+2] * m_dimensions[i+1];
132 m_outputStrides[0] = m_outputStrides[1] * m_dimensions[0]
    [all...]
TensorBroadcasting.h 131 m_outputStrides[0] = 1;
134 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
138 m_outputStrides[NumDims-1] = 1;
141 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
175 const Index idx = index / m_outputStrides[i];
186 index -= idx * m_outputStrides[i];
205 const Index idx = index / m_outputStrides[i];
216 index -= idx * m_outputStrides[i]
    [all...]
TensorInflation.h 114 m_outputStrides[0] = 1;
117 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
121 m_outputStrides[NumDims-1] = 1;
124 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
148 const Index idx = index / m_outputStrides[i];
153 index -= idx * m_outputStrides[i];
162 const Index idx = index / m_outputStrides[i];
167 index -= idx * m_outputStrides[i]
    [all...]
TensorShuffling.h 132 m_outputStrides[0] = 1;
135 m_outputStrides[i] = m_outputStrides[i - 1] * m_dimensions[i - 1];
139 m_outputStrides[NumDims - 1] = 1;
142 m_outputStrides[i] = m_outputStrides[i + 1] * m_dimensions[i + 1];
195 const Index idx = index / m_outputStrides[i];
197 index -= idx * m_outputStrides[i];
202 const Index idx = index / m_outputStrides[i];
204 index -= idx * m_outputStrides[i]
    [all...]
TensorMorphing.h 337 m_outputStrides[0] = 1;
339 m_outputStrides[i] = m_outputStrides[i-1] * output_dims[i-1];
340 m_fastOutputStrides[i] = internal::TensorIntDivisor<Index>(m_outputStrides[i]);
349 m_outputStrides[NumDims-1] = 1;
351 m_outputStrides[i] = m_outputStrides[i+1] * output_dims[i+1];
352 m_fastOutputStrides[i] = internal::TensorIntDivisor<Index>(m_outputStrides[i]);
423 indices[0] -= idx0 * m_outputStrides[i];
424 indices[1] -= idx1 * m_outputStrides[i]
    [all...]
TensorConcatenation.h 157 m_outputStrides[0] = 1;
162 m_outputStrides[j] = m_outputStrides[j-1] * m_dimensions[j-1];
167 m_outputStrides[NumDims - 1] = 1;
172 m_outputStrides[j] = m_outputStrides[j+1] * m_dimensions[j+1];
201 subs[i] = index / m_outputStrides[i];
202 index -= subs[i] * m_outputStrides[i];
207 subs[i] = index / m_outputStrides[i];
208 index -= subs[i] * m_outputStrides[i]
    [all...]
TensorReduction.h 434 m_outputStrides[0] = 1;
436 m_outputStrides[i] = m_outputStrides[i - 1] * m_dimensions[i - 1];
439 m_outputStrides.back() = 1;
441 m_outputStrides[i] = m_outputStrides[i + 1] * m_dimensions[i + 1];
711 const Index idx = index / m_outputStrides[i];
713 index -= idx * m_outputStrides[i];
724 const Index idx = index / m_outputStrides[i];
726 index -= idx * m_outputStrides[i]
    [all...]
TensorConvolution.h 86 m_outputStrides[i] = outputStrides[ordering[i]];
145 outputIndex += idx * m_outputStrides[d];
148 outputIndex += p * m_outputStrides[NumKernelDims];
156 outputIndex += idx * m_outputStrides[d];
159 outputIndex += p * m_outputStrides[limit];
175 return i * m_outputStrides[offset];
189 return i * m_outputStrides[offset] + j * m_outputStrides[offset + 1];
204 return i * m_outputStrides[offset] + j * m_outputStrides[offset + 1]
    [all...]

Completed in 220 milliseconds