Lines Matching refs:TensorMap
14 using Eigen::TensorMap;
66 const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, sizes); in typeCasting()
67 TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes); in typeCasting()
86 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in random()
105 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in slicing()
106 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in slicing()
107 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in slicing()
146 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in rowChip()
149 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in rowChip()
167 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in colChip()
170 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in colChip()
189 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in shuffling()
193 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in shuffling()
216 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in padding()
220 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in padding()
248 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in striding()
252 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in striding()
282 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in broadcasting()
286 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, size_c); in broadcasting()
317 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in coeffWiseOp()
318 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in coeffWiseOp()
319 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in coeffWiseOp()
339 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in algebraicFunc()
340 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in algebraicFunc()
341 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in algebraicFunc()
362 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes); in transcendentalFunc()
363 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes); in transcendentalFunc()
364 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in transcendentalFunc()
384 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in rowReduction()
387 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in rowReduction()
416 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B( in colReduction()
420 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> A( in colReduction()
450 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B( in fullReduction()
453 TensorMap<Tensor<T, 0, 0, TensorIndex>, Eigen::Aligned> C( in fullReduction()
496 TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, input_sizes); in convolution()
500 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes); in convolution()
504 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, result_sizes); in convolution()
537 const TensorMap<Tensor<T, 2, Layout>, Eigen::Aligned> A(a_, sizeA); in contraction()
538 const TensorMap<Tensor<T, 2, Layout>, Eigen::Aligned> B(b_, sizeB); in contraction()
539 TensorMap<Tensor<T, 2, Layout>, Eigen::Aligned> C(c_, sizeC); in contraction()