| /external/tensorflow/tensorflow/core/kernels/ |
| topk_op_gpu.cu.cc | 520 const Eigen::DSizes<Eigen::DenseIndex, 2> slice_indices{0, 0};
521 const Eigen::DSizes<Eigen::DenseIndex, 2> slice_sizes{num_rows, k};
|
| quantized_concat_op.cc | 47 Eigen::array<Eigen::DenseIndex, 1> dims;
|
| tensor_array_ops.cc | [all...] |
| cwise_ops.h | 977 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast0,
979 typename Eigen::array<Eigen::DenseIndex, NDIMS> bcast1,
991 bool AllOne(const typename Eigen::array<Eigen::DenseIndex, NDIMS>& a) {
[all...] |
| adjust_contrast_op.cc | 242 typedef Eigen::array<Eigen::DenseIndex, 1> Index;
|
| conv_2d.h | 135 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
|
| batch_kernels.cc | 160 Eigen::DSizes<Eigen::DenseIndex, 3> slice_indices{0, position, 0};
161 Eigen::DSizes<Eigen::DenseIndex, 3> slice_sizes{1, size, suffix_dim_size};
[all...] |
| conv_ops_gpu_3.cu.cc | [all...] |
| eigen_backward_spatial_convolutions_test.cc | [all...] |
| mkl_lrn_op.cc | 62 Eigen::DSizes<Eigen::DenseIndex, 2> start(row, begin);
63 Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, end - begin);
[all...] |
| pooling_ops_common.h | 575 Eigen::DSizes<Eigen::DenseIndex, 2> in_indices(0, in_offset);
|
| sparse_matmul_op.cc | 62 using DSizes = Eigen::DSizes<Eigen::DenseIndex, 2>;
[all...] |
| conv_grad_filter_ops.cc | 451 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims;
[all...] |
| conv_grad_input_ops.cc | 494 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims;
[all...] |
| quantization_utils.h | [all...] |
| /external/eigen/unsupported/test/ |
| cxx11_tensor_ref.cpp | 209 DSizes<Eigen::DenseIndex, 4> shuffle_dims(0, 1, 2, 3);
|
| cxx11_tensor_simple.cpp | 198 array<Eigen::DenseIndex, 3> dims;
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorEvaluator.h | 88 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(const array<DenseIndex, NumCoords>& coords) const {
97 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar& coeffRef(const array<DenseIndex, NumCoords>& coords) {
198 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(const array<DenseIndex, NumCoords>& coords) const {
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops.cc | 65 Eigen::array<Eigen::DenseIndex, 2> b_shape({1, b.dimensions()[0]});
66 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({cell.batch_size(), 1});
69 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()});
70 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({cell.batch_size(), 1});
148 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()});
149 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({cell.batch_size(), 1});
[all...] |
| /external/eigen/Eigen/src/Core/ |
| DenseBase.h | 21 EIGEN_STATIC_ASSERT(NumTraits<DenseIndex>::IsSigned,THE_INDEX_TYPE_MUST_BE_A_SIGNED_TYPE);
|
| /external/eigen/Eigen/src/Core/util/ |
| Meta.h | 25 typedef EIGEN_DEFAULT_DENSE_INDEX_TYPE DenseIndex;
|
| /external/eigen/unsupported/Eigen/src/NonLinearOptimization/ |
| HybridNonLinearSolver.h | 46 typedef DenseIndex Index;
|
| LevenbergMarquardt.h | 58 typedef DenseIndex Index;
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_test.cc | 847 Eigen::DSizes<Eigen::DenseIndex, 2> indices(i, 0);
848 Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, 4 * 3 * 2);
[all...] |
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor.h | 473 Eigen::DenseIndex offset = 0;
|
