Lines Matching refs:TensorMap
14 using Eigen::TensorMap;
56 const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, sizes);
57 TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes);
72 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
87 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes);
88 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
89 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
117 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
120 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size);
134 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
137 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size);
152 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a);
156 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b);
175 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a);
179 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b);
203 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a);
207 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b);
231 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, size_a);
235 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, size_c);
261 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes);
262 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
263 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
279 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes);
280 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
281 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
297 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes);
298 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
299 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
315 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
318 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size);
343 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(
347 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(
373 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(
376 TensorMap<Tensor<T, 0, 0, TensorIndex>, Eigen::Aligned> C(
400 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizeA);
401 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizeB);
402 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizeC);
421 TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, input_sizes);
425 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes);
429 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, result_sizes);
