Lines Matching refs:Eigen
13 using Eigen::Tensor;
14 using Eigen::TensorMap;
48 Eigen::array<TensorIndex, 2> sizes; in typeCasting()
56 const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, sizes); in typeCasting()
57 TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes); in typeCasting()
69 Eigen::array<TensorIndex, 2> sizes; in random()
72 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in random()
84 Eigen::array<TensorIndex, 2> sizes; in slicing()
87 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in slicing()
88 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in slicing()
89 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in slicing()
91 const Eigen::DSizes<TensorIndex, 2> quarter_sizes(m_/2, m_/2); in slicing()
92 const Eigen::DSizes<TensorIndex, 2> first_quadrant(0, 0); in slicing()
93 const Eigen::DSizes<TensorIndex, 2> second_quadrant(0, m_/2); in slicing()
94 const Eigen::DSizes<TensorIndex, 2> third_quadrant(m_/2, 0); in slicing()
95 const Eigen::DSizes<TensorIndex, 2> fourth_quadrant(m_/2, m_/2); in slicing()
114 Eigen::array<TensorIndex, 2> input_size; in rowChip()
117 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in rowChip()
118 Eigen::array<TensorIndex, 1> output_size; in rowChip()
120 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in rowChip()
131 Eigen::array<TensorIndex, 2> input_size; in colChip()
134 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in colChip()
135 Eigen::array<TensorIndex, 1> output_size; in colChip()
137 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in colChip()
149 Eigen::array<TensorIndex, 2> size_a; in shuffling()
152 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in shuffling()
153 Eigen::array<TensorIndex, 2> size_b; in shuffling()
156 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in shuffling()
158 Eigen::array<int, 2> shuffle; in shuffling()
172 Eigen::array<TensorIndex, 2> size_a; in padding()
175 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in padding()
176 Eigen::array<TensorIndex, 2> size_b; in padding()
179 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in padding()
182 Eigen::IndexPairList<Eigen::type2indexpair<0, 0>, in padding()
183 Eigen::type2indexpair<2, 1> > paddings; in padding()
185 Eigen::array<Eigen::IndexPair<TensorIndex>, 2> paddings; in padding()
186 paddings[0] = Eigen::IndexPair<TensorIndex>(0, 0); in padding()
187 paddings[1] = Eigen::IndexPair<TensorIndex>(2, 1); in padding()
200 Eigen::array<TensorIndex, 2> size_a; in striding()
203 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in striding()
204 Eigen::array<TensorIndex, 2> size_b; in striding()
207 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in striding()
210 Eigen::array<TensorIndex, 2> strides; in striding()
216 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> > strides; in striding()
228 Eigen::array<TensorIndex, 2> size_a; in broadcasting()
231 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in broadcasting()
232 Eigen::array<TensorIndex, 2> size_c; in broadcasting()
235 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, size_c); in broadcasting()
238 Eigen::array<int, 2> broadcast; in broadcasting()
244 Eigen::IndexList<Eigen::type2index<1>, int> broadcast; in broadcasting()
258 Eigen::array<TensorIndex, 2> sizes; in coeffWiseOp()
261 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in coeffWiseOp()
262 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in coeffWiseOp()
263 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in coeffWiseOp()
276 Eigen::array<TensorIndex, 2> sizes; in algebraicFunc()
279 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in algebraicFunc()
280 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in algebraicFunc()
281 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in algebraicFunc()
294 Eigen::array<TensorIndex, 2> sizes; in transcendentalFunc()
297 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in transcendentalFunc()
298 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in transcendentalFunc()
299 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in transcendentalFunc()
312 Eigen::array<TensorIndex, 2> input_size; in rowReduction()
315 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in rowReduction()
316 Eigen::array<TensorIndex, 1> output_size; in rowReduction()
318 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in rowReduction()
321 Eigen::array<TensorIndex, 1> sum_along_dim; in rowReduction()
326 Eigen::IndexList<Eigen::type2index<0>> sum_along_dim; in rowReduction()
340 Eigen::array<TensorIndex, 2> input_size; in colReduction()
343 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B( in colReduction()
345 Eigen::array<TensorIndex, 1> output_size; in colReduction()
347 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C( in colReduction()
351 Eigen::array<TensorIndex, 1> sum_along_dim; in colReduction()
356 Eigen::IndexList<Eigen::type2index<1>> sum_along_dim; in colReduction()
370 Eigen::array<TensorIndex, 2> input_size; in fullReduction()
373 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B( in fullReduction()
375 Eigen::array<TensorIndex, 0> output_size; in fullReduction()
376 TensorMap<Tensor<T, 0, 0, TensorIndex>, Eigen::Aligned> C( in fullReduction()
390 Eigen::array<TensorIndex, 2> sizeA; in contraction()
393 Eigen::array<TensorIndex, 2> sizeB; in contraction()
396 Eigen::array<TensorIndex, 2> sizeC; in contraction()
400 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizeA); in contraction()
401 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizeB); in contraction()
402 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizeC); in contraction()
405 Eigen::array<DimPair, 1> dims; in contraction()
418 Eigen::array<TensorIndex, 2> input_sizes; in convolution()
421 TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, input_sizes); in convolution()
422 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution()
425 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes); in convolution()
426 Eigen::array<TensorIndex, 2> result_sizes; in convolution()
429 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, result_sizes); in convolution()
430 Eigen::array<TensorIndex, 2> dims; in convolution()
461 if (Eigen::internal::is_same<Device, Eigen::GpuDevice>::value) { in finalizeBenchmark()