| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_tensor_dense_add_op.cc | 130 Eigen::array<Eigen::DenseIndex, NDIMS> idx;
|
| cwise_ops_gpu_common.cu.h | 81 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast0,
83 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast1,
|
| cwise_ops_sycl_common.h | 84 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast0,
86 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast1,
|
| pooling_ops_3d.cc | 248 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices;
249 Eigen::DSizes<Eigen::DenseIndex, 5> src_sizes;
250 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices;
251 Eigen::DSizes<Eigen::DenseIndex, 5> dst_sizes;
423 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices;
424 Eigen::DSizes<Eigen::DenseIndex, 5> src_sizes;
425 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices;
426 Eigen::DSizes<Eigen::DenseIndex, 5> dst_sizes;
[all...] |
| summary_image_op.cc | 87 &values(i, 0, 0), Eigen::DSizes<Eigen::DenseIndex, 2>(hw, depth));
126 Eigen::DSizes<Eigen::DenseIndex, 2>(hw, depth));
|
| cwise_ops_common.h | 311 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast0,
313 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast1,
387 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast0,
389 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast1,
507 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast0,
509 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast1,
|
| conv_ops.cc | 80 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
81 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
96 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
97 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
[all...] |
| segment_reduction_ops.cc | 115 Eigen::DSizes<Eigen::DenseIndex, 1> dims_to_reduce;
127 Eigen::DSizes<Eigen::DenseIndex, 1> out_slice_shape(num_col);
159 Eigen::DSizes<Eigen::DenseIndex, 2> gap_slice_shape(
179 Eigen::DSizes<Eigen::DenseIndex, 2> in_slice_shape(end - start,
717 Eigen::DSizes<Eigen::DenseIndex, 2> gap_slice_shape(
[all...] |
| padding_fifo_queue.cc | 323 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_indices;
325 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_size;
|
| sparse_conditional_accumulator.h | 270 Eigen::DSizes<Eigen::DenseIndex, 1> slice_shape(num_col);
339 Eigen::DSizes<Eigen::DenseIndex, 1> slice_shape(accum_flat.dimension(1));
|
| mkl_matmul_op.cc | 55 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
|
| quantized_matmul_op.cc | 101 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
|
| lrn_op.cc | 57 Eigen::DSizes<Eigen::DenseIndex, 2> start(row, begin);
58 Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, end - begin);
|
| scatter_nd_op.cc | 474 typename Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix; \
518 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix, \
|
| serialize_sparse_op.cc | 370 Eigen::DSizes<Eigen::DenseIndex, 2> indices_start(0, 1);
371 Eigen::DSizes<Eigen::DenseIndex, 2> indices_sizes(num_entries, rank);
|
| sparse_tensors_map_ops.cc | 423 Eigen::DSizes<Eigen::DenseIndex, 2> indices_start(0, 1);
424 Eigen::DSizes<Eigen::DenseIndex, 2> indices_sizes(num_entries, rank);
|
| /external/eigen/bench/btl/libs/eigen3/ |
| eigen3_interface.hh | 219 Matrix<DenseIndex,1,Dynamic> piv(N);
220 DenseIndex nb;
|
| /external/eigen/unsupported/Eigen/src/NonLinearOptimization/ |
| lmpar.h | 17 typedef DenseIndex Index;
175 typedef DenseIndex Index;
|
| /external/eigen/unsupported/Eigen/src/Polynomials/ |
| Companion.h | 60 typedef DenseIndex Index;
|
| PolynomialSolver.h | 39 typedef DenseIndex Index;
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor_test.cc | 34 Eigen::Tensor<int64, 2, Eigen::RowMajor, Eigen::DenseIndex>
36 Eigen::Tensor<int64, 2, Eigen::RowMajor, Eigen::DenseIndex> ix(N, NDIM);
360 Eigen::array<Eigen::DenseIndex, NDIM> ix_n;
399 Eigen::array<Eigen::DenseIndex, NDIM> ix_n;
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorFunctors.h | 182 DenseIndex scalarCount_;
183 DenseIndex packetCount_;
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops_gpu.cu.cc | 338 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size});
339 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({batch_size, 1});
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_test.cc | 388 const Eigen::array<Eigen::DenseIndex, 1> sum_along_dim(0);
389 const Eigen::array<Eigen::DenseIndex, 2> matrix_transpose({1, 0});
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_shape_test.cc | 287 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizes() const;
292 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizesWithPadding() const;
|
