• Home
  • Raw
  • Download

Lines Matching refs:Aligned

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()
72 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes); in random()
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()
117 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in rowChip()
120 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in rowChip()
134 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in colChip()
137 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in colChip()
152 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in shuffling()
156 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in shuffling()
175 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in padding()
179 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in padding()
203 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in striding()
207 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); in striding()
231 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a); in broadcasting()
235 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, size_c); in broadcasting()
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()
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()
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()
315 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size); in rowReduction()
318 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size); in rowReduction()
343 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B( in colReduction()
347 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C( in colReduction()
373 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B( in fullReduction()
376 TensorMap<Tensor<T, 0, 0, TensorIndex>, Eigen::Aligned> C( in fullReduction()
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()
421 TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, input_sizes); in convolution()
425 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes); in convolution()
429 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, result_sizes); in convolution()