Lines Matching refs:TensorIndex
4 typedef int TensorIndex; typedef
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()
84 Eigen::array<TensorIndex, 2> 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()
153 Eigen::array<TensorIndex, 2> size_b; in shuffling()
172 Eigen::array<TensorIndex, 2> size_a; in padding()
176 Eigen::array<TensorIndex, 2> size_b; 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()
204 Eigen::array<TensorIndex, 2> size_b; in striding()
210 Eigen::array<TensorIndex, 2> strides; in striding()
228 Eigen::array<TensorIndex, 2> size_a; in broadcasting()
232 Eigen::array<TensorIndex, 2> size_c; in broadcasting()
258 Eigen::array<TensorIndex, 2> sizes; in coeffWiseOp()
276 Eigen::array<TensorIndex, 2> sizes; in algebraicFunc()
294 Eigen::array<TensorIndex, 2> 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()
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()
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()
418 Eigen::array<TensorIndex, 2> input_sizes; in convolution()
422 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution()
426 Eigen::array<TensorIndex, 2> result_sizes; in convolution()
430 Eigen::array<TensorIndex, 2> dims; in convolution()
470 TensorIndex m_;
471 TensorIndex k_;
472 TensorIndex n_;