| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorMap.h | 240 const std::size_t NumDims = sizeof...(otherIndices) + 2;
242 const Index index = m_dimensions.IndexOfRowMajor(array<Index, NumDims>{{firstIndex, secondIndex, otherIndices...}});
245 const Index index = m_dimensions.IndexOfColMajor(array<Index, NumDims>{{firstIndex, secondIndex, otherIndices...}});
|
| TensorArgMax.h | 84 static const int NumDims = internal::array_size<Dimensions>::value;
210 static const int NumDims = internal::array_size<InputDimensions>::value;
211 typedef array<Index, NumDims> StrideDims;
230 m_stride_mod = (m_return_dim < NumDims - 1) ? m_strides[m_return_dim + 1] : total_size;
270 eigen_assert(m_return_dim < NumDims &&
277 for (int i = 1; i < NumDims; ++i) {
281 strides[NumDims-1] = 1;
282 for (int i = NumDims - 2; i >= 0; --i) {
|
| TensorBroadcasting.h | 100 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
101 typedef DSizes<Index, NumDims> Dimensions;
121 EIGEN_STATIC_ASSERT((NumDims > 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
124 for (int i = 0; i < NumDims; ++i) {
132 for (int i = 1; i < NumDims; ++i) {
137 m_inputStrides[NumDims-1] = 1;
138 m_outputStrides[NumDims-1] = 1;
139 for (int i = NumDims-2; i >= 0; --i) {
174 for (int i = NumDims - 1; i > 0; --i) {
204 for (int i = 0; i < NumDims - 1; ++i)
[all...] |
| TensorCustomOp.h | 84 static const int NumDims = internal::traits<ArgType>::NumDimensions;
85 typedef DSizes<Index, NumDims> Dimensions;
145 TensorMap<Tensor<CoeffReturnType, NumDims, Layout, Index> > result(
240 static const int NumDims = internal::traits<XprType>::NumDimensions;
241 typedef DSizes<Index, NumDims> Dimensions;
300 TensorMap<Tensor<Scalar, NumDims, Layout> > result(data, m_dimensions);
|
| TensorGenerator.h | 86 static const int NumDims = internal::array_size<Dimensions>::value;
107 for (int i = 1; i < NumDims; ++i) {
111 m_strides[NumDims - 1] = 1;
112 for (int i = NumDims - 2; i >= 0; --i) {
128 array<Index, NumDims> coords;
160 void extract_coordinates(Index index, array<Index, NumDims>& coords) const {
162 for (int i = NumDims - 1; i > 0; --i) {
169 for (int i = 0; i < NumDims - 1; ++i) {
174 coords[NumDims-1] = index;
179 array<Index, NumDims> m_strides
[all...] |
| TensorInflation.h | 82 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
83 typedef DSizes<Index, NumDims> Dimensions;
103 for (int i = 0; i < NumDims; ++i) {
108 for (int i = 0; i < NumDims; ++i) {
116 for (int i = 1; i < NumDims; ++i) {
121 m_outputStrides[NumDims-1] = 1;
122 m_inputStrides[NumDims-1] = 1;
123 for (int i = NumDims - 2; i >= 0; --i) {
147 for (int i = NumDims - 1; i > 0; --i) {
161 for (int i = 0; i < NumDims - 1; ++i)
[all...] |
| TensorLayoutSwap.h | 115 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
116 typedef DSizes<Index, NumDims> Dimensions;
129 for(int i = 0; i < NumDims; ++i) {
130 m_dimensions[i] = m_impl.dimensions()[NumDims-1-i];
|
| TensorPatch.h | 85 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value + 1;
86 typedef DSizes<Index, NumDims> Dimensions;
108 for (int i = 0; i < NumDims-1; ++i) {
112 m_dimensions[NumDims-1] = num_patches;
116 for (int i = 1; i < NumDims-1; ++i) {
121 for (int i = 1; i < NumDims; ++i) {
125 for (int i = 0; i < NumDims-1; ++i) {
131 m_inputStrides[NumDims-2] = 1;
132 m_patchStrides[NumDims-2] = 1;
133 for (int i = NumDims-3; i >= 0; --i)
[all...] |
| TensorReverse.h | 105 static const int NumDims = internal::array_size<ReverseDimensions>::value;
106 typedef DSizes<Index, NumDims> Dimensions;
125 EIGEN_STATIC_ASSERT((NumDims > 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
131 for (int i = 1; i < NumDims; ++i) {
135 m_strides[NumDims-1] = 1;
136 for (int i = NumDims - 2; i >= 0; --i) {
158 for (int i = NumDims - 1; i > 0; --i) {
172 for (int i = 0; i < NumDims - 1; ++i) {
180 if (m_reverse[NumDims-1]) {
181 inputIndex += (m_dimensions[NumDims-1] - index - 1)
[all...] |
| TensorShuffling.h | 104 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
105 typedef DSizes<Index, NumDims> Dimensions;
124 for (int i = 0; i < NumDims; ++i) {
128 array<Index, NumDims> inputStrides;
133 for (int i = 1; i < NumDims; ++i) {
138 inputStrides[NumDims - 1] = 1;
139 m_outputStrides[NumDims - 1] = 1;
140 for (int i = NumDims - 2; i >= 0; --i) {
146 for (int i = 0; i < NumDims; ++i) {
181 const double compute_cost = NumDims * (2 * TensorOpCost::AddCost<Index>()
[all...] |
| TensorStriding.h | 104 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
105 typedef DSizes<Index, NumDims> Dimensions;
123 for (int i = 0; i < NumDims; ++i) {
131 for (int i = 1; i < NumDims; ++i) {
136 m_inputStrides[NumDims-1] *= op.strides()[NumDims-1];
138 m_outputStrides[NumDims-1] = 1;
139 m_inputStrides[NumDims-1] = 1;
140 for (int i = NumDims - 2; i >= 0; --i) {
173 for (int i = NumDims - 1; i > 0; --i)
[all...] |
| TensorChipping.h | 133 static const int NumDims = NumInputDims-1;
135 typedef DSizes<Index, NumDims> Dimensions;
268 if (((static_cast<int>(Layout) == static_cast<int>(ColMajor) && m_dim.actualDim() == NumDims) ||
318 static const int NumDims = NumInputDims-1;
320 typedef DSizes<Index, NumDims> Dimensions;
|
| TensorConcatenation.h | 114 static const int NumDims = internal::array_size<typename TensorEvaluator<LeftArgType, Device>::Dimensions>::value;
116 typedef DSizes<Index, NumDims> Dimensions;
130 EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<LeftArgType, Device>::Layout) == static_cast<int>(TensorEvaluator<RightArgType, Device>::Layout) || NumDims == 1), YOU_MADE_A_PROGRAMMING_MISTAKE);
131 EIGEN_STATIC_ASSERT((NumDims == RightNumDims), YOU_MADE_A_PROGRAMMING_MISTAKE);
132 EIGEN_STATIC_ASSERT((NumDims > 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
134 eigen_assert(0 <= m_axis && m_axis < NumDims);
147 for (++i; i < NumDims; ++i) {
159 for (int j = 1; j < NumDims; ++j) {
165 m_leftStrides[NumDims - 1] = 1;
166 m_rightStrides[NumDims - 1] = 1
[all...] |
| TensorPadding.h | 88 static const int NumDims = internal::array_size<PaddingDimensions>::value;
89 typedef DSizes<Index, NumDims> Dimensions;
109 EIGEN_STATIC_ASSERT((NumDims > 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
113 for (int i = 0; i < NumDims; ++i) {
120 for (int i = 1; i < NumDims; ++i) {
124 m_outputStrides[NumDims] = m_outputStrides[NumDims-1] * m_dimensions[NumDims-1];
126 m_inputStrides[NumDims - 1] = 1;
127 m_outputStrides[NumDims] = 1
[all...] |
| TensorScan.h | 88 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
89 typedef DSizes<Index, NumDims> Dimensions;
115 EIGEN_STATIC_ASSERT((NumDims > 0), YOU_MADE_A_PROGRAMMING_MISTAKE);
116 eigen_assert(op.axis() >= 0 && op.axis() < NumDims);
125 for (int i = NumDims - 1; i > op.axis(); --i) {
|
| TensorFFT.h | 121 static const int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
122 typedef DSizes<Index, NumDims> Dimensions;
145 for (int i = 0; i < NumDims; ++i) {
152 for (int i = 1; i < NumDims; ++i) {
156 m_strides[NumDims - 1] = 1;
157 for (int i = NumDims - 2; i >= 0; --i) {
217 eigen_assert(dim >= 0 && dim < NumDims);
537 for (int i = NumDims - 1; i > omitted_dim; --i) {
567 array<Index, NumDims> m_strides;
|
| TensorMorphing.h | 309 static const int NumDims = internal::array_size<Sizes>::value;
332 for (int i = 1; i < NumDims; ++i) {
338 for (int i = 1; i < NumDims; ++i) {
343 m_inputStrides[NumDims-1] = 1;
344 for (int i = NumDims - 2; i >= 0; --i) {
348 // Don't initialize m_fastOutputStrides[NumDims-1] since it won't ever be accessed.
349 m_outputStrides[NumDims-1] = 1;
350 for (int i = NumDims - 2; i >= 0; --i) {
371 for (int i = 0; i < NumDims; ++i) {
378 for (int i = NumDims-1; i >= 0; --i)
[all...] |
| TensorContraction.h | 40 // From NumDims below.
68 // From NumDims below.
144 static const int NumDims = LDims + RDims - 2 * ContractDims;
150 typedef DSizes<Index, NumDims> Dimensions;
322 for (int i = 0, j = NumDims - 1; i < j; i++, j--) {
607 static const int NumDims = LDims + RDims - 2 * ContractDims;
610 typedef DSizes<Index, NumDims> Dimensions;
|
| TensorContractionCuda.h | [all...] |
| TensorConvolution.h | 30 array<Index, NumDims> dimensions = input_dims;
39 array<Index, NumDims> inputStrides;
40 array<Index, NumDims> outputStrides;
44 for (int i = 1; i < NumDims; ++i) {
49 inputStrides[NumDims - 1] = 1;
50 outputStrides[NumDims - 1] = 1;
51 for (int i = static_cast<int>(NumDims) - 2; i >= 0; --i) {
57 array<Index, NumDims> cudaInputDimensions;
58 array<Index, NumDims> cudaOutputDimensions;
59 array<Index, NumDims> tmp = dimensions
[all...] |
| TensorImagePatch.h | 156 static const int NumDims = NumInputDims + 1;
157 typedef DSizes<Index, NumDims> Dimensions;
177 EIGEN_STATIC_ASSERT((NumDims >= 4), YOU_MADE_A_PROGRAMMING_MISTAKE);
261 for (int i = 4; i < NumDims; ++i) {
266 // NumDims-1: depth
267 // NumDims-2: patch_rows
268 // NumDims-3: patch_cols
269 // NumDims-4: number of patches
270 // NumDims-5 and beyond: anything else (such as batch).
271 m_dimensions[NumDims-1] = input_dims[NumInputDims-1]
[all...] |
| TensorContractionThreadPool.h | 100 static const int NumDims = LDims + RDims - 2 * ContractDims;
102 typedef DSizes<Index, NumDims> Dimensions;
[all...] |
| TensorVolumePatch.h | 171 static const int NumDims = NumInputDims + 1;
172 typedef DSizes<Index, NumDims> Dimensions;
190 EIGEN_STATIC_ASSERT((NumDims >= 5), YOU_MADE_A_PROGRAMMING_MISTAKE);
281 for (int i = 5; i < NumDims; ++i) {
286 // NumDims-1: depth
287 // NumDims-2: patch_planes
288 // NumDims-3: patch_rows
289 // NumDims-4: patch_cols
290 // NumDims-5: number of patches
291 // NumDims-6 and beyond: anything else (such as batch)
[all...] |
