Home | History | Annotate | Download | only in tensors

Lines Matching refs:array

48     Eigen::array<TensorIndex, 2> sizes;
69 Eigen::array<TensorIndex, 2> sizes;
84 Eigen::array<TensorIndex, 2> sizes;
114 Eigen::array<TensorIndex, 2> input_size;
118 Eigen::array<TensorIndex, 1> output_size;
131 Eigen::array<TensorIndex, 2> input_size;
135 Eigen::array<TensorIndex, 1> output_size;
149 Eigen::array<TensorIndex, 2> size_a;
153 Eigen::array<TensorIndex, 2> size_b;
158 Eigen::array<int, 2> shuffle;
172 Eigen::array<TensorIndex, 2> size_a;
176 Eigen::array<TensorIndex, 2> size_b;
185 Eigen::array<Eigen::IndexPair<TensorIndex>, 2> paddings;
200 Eigen::array<TensorIndex, 2> size_a;
204 Eigen::array<TensorIndex, 2> size_b;
210 Eigen::array<TensorIndex, 2> strides;
228 Eigen::array<TensorIndex, 2> size_a;
232 Eigen::array<TensorIndex, 2> size_c;
238 Eigen::array<int, 2> broadcast;
258 Eigen::array<TensorIndex, 2> sizes;
276 Eigen::array<TensorIndex, 2> sizes;
294 Eigen::array<TensorIndex, 2> sizes;
312 Eigen::array<TensorIndex, 2> input_size;
316 Eigen::array<TensorIndex, 1> output_size;
321 Eigen::array<TensorIndex, 1> sum_along_dim;
340 Eigen::array<TensorIndex, 2> input_size;
345 Eigen::array<TensorIndex, 1> output_size;
351 Eigen::array<TensorIndex, 1> sum_along_dim;
370 Eigen::array<TensorIndex, 2> input_size;
375 Eigen::array<TensorIndex, 0> output_size;
390 Eigen::array<TensorIndex, 2> sizeA;
393 Eigen::array<TensorIndex, 2> sizeB;
396 Eigen::array<TensorIndex, 2> sizeC;
405 Eigen::array<DimPair, 1> dims;
418 Eigen::array<TensorIndex, 2> input_sizes;
422 Eigen::array<TensorIndex, 2> kernel_sizes;
426 Eigen::array<TensorIndex, 2> result_sizes;
430 Eigen::array<TensorIndex, 2> dims;