| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| eigen_spatial_convolutions.h | 98 const DenseIndex row_stride = 1,
99 const DenseIndex col_stride = 1,
101 const DenseIndex row_in_stride = 1,
102 const DenseIndex col_in_stride = 1) {
132 const DenseIndex kernelRowsEff =
134 const DenseIndex kernelColsEff =
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops.h | 86 Eigen::array<Eigen::DenseIndex, 2> m_shape({m.dimensions()[0], 1});
88 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({1, m.dimensions()[1]});
107 inline Eigen::array<Eigen::DenseIndex, 2> icfo_i_offsets() const {
111 inline Eigen::array<Eigen::DenseIndex, 2> icfo_c_offsets() const {
115 inline Eigen::array<Eigen::DenseIndex, 2> icfo_f_offsets() const {
119 inline Eigen::array<Eigen::DenseIndex, 2> icfo_o_offsets() const {
123 inline Eigen::array<Eigen::DenseIndex, 2> cell_extents() const {
127 inline Eigen::array<Eigen::DenseIndex, 2> xh_x_offsets() const {
131 inline Eigen::array<Eigen::DenseIndex, 2> xh_x_extents() const {
135 inline Eigen::array<Eigen::DenseIndex, 2> xh_h_offsets() const
[all...] |
| blas_gemm.h | 60 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
62 Eigen::IndexPair<Eigen::DenseIndex>(transa == false, transb == true);
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorChipping.h | 24 template<DenseIndex DimId, typename XprType>
37 template<DenseIndex DimId, typename XprType>
43 template<DenseIndex DimId, typename XprType>
49 template <DenseIndex DimId>
52 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DimensionId(DenseIndex dim) {
55 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const {
62 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DimensionId(DenseIndex dim) : actual_dim(dim) {
65 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const {
69 const DenseIndex actual_dim;
77 template<DenseIndex DimId, typename XprType
[all...] |
| TensorStorage.h | 64 EIGEN_STRONG_INLINE DenseIndex size() const { return m_dimensions.TotalSize(); }
89 template <typename... DenseIndex>
90 EIGEN_DEVICE_FUNC TensorStorage(DenseIndex... indices) : m_dimensions(indices...) {
|
| TensorRef.h | 26 EIGEN_DEVICE_FUNC virtual const Scalar coeff(DenseIndex index) const = 0;
27 EIGEN_DEVICE_FUNC virtual Scalar& coeffRef(DenseIndex index) = 0;
63 EIGEN_DEVICE_FUNC virtual const Scalar coeff(DenseIndex index) const {
66 EIGEN_DEVICE_FUNC virtual Scalar& coeffRef(DenseIndex /*index*/) {
88 EIGEN_DEVICE_FUNC virtual Scalar& coeffRef(DenseIndex index) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| unpack_op.cc | 70 std::numeric_limits<Eigen::DenseIndex>::max()),
71 errors::InvalidArgument("output size must fit in Eigen DenseIndex"));
116 Eigen::DSizes<Eigen::DenseIndex, 3> indices{0, 0, i * after_dim};
117 Eigen::DSizes<Eigen::DenseIndex, 3> sizes{1, before_dim, after_dim};
|
| one_hot_op.h | 40 operator()(const Eigen::array<Eigen::DenseIndex, 3>& pre_depth_suff) const {
|
| slice_op.cc | 218 Eigen::DSizes<Eigen::DenseIndex, NDIM> indices;
219 Eigen::DSizes<Eigen::DenseIndex, NDIM> sizes;
402 Eigen::DSizes<Eigen::DenseIndex, NDIM> indices;
403 Eigen::DSizes<Eigen::DenseIndex, NDIM> sizes;
424 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, \
425 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes); \
477 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, \
478 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes); \
539 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, \
540 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes);
[all...] |
| tile_ops.cc | 59 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices,
60 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes,
68 const Eigen::DSizes<Eigen::DenseIndex, 0>&,
69 const Eigen::DSizes<Eigen::DenseIndex, 0>&, bool first) const;
77 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
78 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& reshape_dim) const;
381 Eigen::DSizes<Eigen::DenseIndex, NDIM> indices;
382 Eigen::DSizes<Eigen::DenseIndex, NDIM> sizes;
418 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim;
419 Eigen::DSizes<Eigen::DenseIndex, NDIM> reshape_dim
[all...] |
| batch_matmul_op_impl.h | 48 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) {
51 return Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
53 return Eigen::IndexPair<Eigen::DenseIndex>(1, 1);
57 return Eigen::IndexPair<Eigen::DenseIndex>(0, 0);
59 return Eigen::IndexPair<Eigen::DenseIndex>(0, 1);
86 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
116 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
416 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
|
| eigen_backward_spatial_convolutions.h | 102 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
103 const DenseIndex row_in_stride = 1, const DenseIndex col_in_stride = 1) {
354 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
355 const DenseIndex row_in_stride = 1, const DenseIndex col_in_stride = 1) {
|
| split_lib_gpu.cu.cc | 36 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices,
37 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes) {
45 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_indices,
46 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_sizes) {
|
| betainc_op.cc | 138 const typename Eigen::array<Eigen::DenseIndex, NDIM>& bcast_a, \
140 const typename Eigen::array<Eigen::DenseIndex, NDIM>& bcast_b, \
142 const typename Eigen::array<Eigen::DenseIndex, NDIM>& bcast_x, \
|
| dynamic_partition_op.cc | 138 Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, slice_size);
151 Eigen::DSizes<Eigen::DenseIndex, 2> out_indices(oi, 0);
152 Eigen::DSizes<Eigen::DenseIndex, 2> data_indices(i, 0);
|
| eigen_cuboid_convolution.h | 73 const DenseIndex stridePlanes = 1,
74 const DenseIndex strideRows = 1,
75 const DenseIndex strideCols = 1,
|
| scatter_nd_op_gpu.cu.cc | 84 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix,
115 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix,
125 const Eigen::DenseIndex batch_size = Tindices.dimension(0);
|
| matmul_op.cc | 77 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
107 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
122 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
260 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
465 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
517 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
530 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
|
| conv_grad_ops_3d.cc | 179 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 5> pad_dims{
196 Eigen::DSizes<Eigen::DenseIndex, 5> no_op_shuffle{0, 1, 2, 3, 4};
197 Eigen::DSizes<Eigen::DenseIndex, 5> eigen_strides{1, strides[0], strides[1],
199 functor::InflatePadAndShuffle<Device, T, 5, Eigen::DenseIndex>()(
211 Eigen::DSizes<Eigen::DenseIndex, 5> filter_order{0, 1, 2, 4, 3};
213 functor::ShuffleAndReverse<Device, T, 5, Eigen::DenseIndex>()(
291 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 5> pad_dims{
312 Eigen::DSizes<Eigen::DenseIndex, 5> out_order{4, 1, 2, 3, 0};
319 Eigen::DSizes<Eigen::DenseIndex, 5> eigen_strides{1, strides[0], strides[1],
321 functor::InflatePadAndShuffle<Device, T, 5, Eigen::DenseIndex>()(
[all...] |
| eigen_spatial_convolutions.h | 27 template <typename NewDimension, DenseIndex Rows, DenseIndex Cols,
456 template <typename NewDimension, DenseIndex Rows, DenseIndex Cols,
626 template <typename NewDimension, DenseIndex Rows, DenseIndex Cols,
[all...] |
| /external/eigen/unsupported/Eigen/src/NonLinearOptimization/ |
| rwupdt.h | 12 typedef DenseIndex Index;
|
| chkder.h | 23 typedef DenseIndex Index;
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor.h | 521 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const;
526 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const;
600 Eigen::array<Eigen::DenseIndex, NDIMS> dims;
616 Eigen::array<Eigen::DenseIndex, NDIMS> dims;
626 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const {
639 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const {
664 Eigen::array<Eigen::DenseIndex, NDIMS> dims;
673 Eigen::array<Eigen::DenseIndex, NDIMS> dims;
682 Eigen::array<Eigen::DenseIndex, NDIMS> dims;
692 Eigen::array<Eigen::DenseIndex, NDIMS> dims
[all...] |
| tensor_shape.h | 298 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizes() const;
303 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizesWithPadding() const;
456 Eigen::DSizes<Eigen::DenseIndex, NDIMS> TensorShape::AsEigenDSizes() const {
462 Eigen::DSizes<Eigen::DenseIndex, NDIMS> TensorShape::AsEigenDSizesWithPadding()
466 Eigen::DSizes<Eigen::DenseIndex, NDIMS> dsizes;
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_morphing.cpp | 123 DSizes<DenseIndex, 1> offsets;
325 typedef Eigen::DSizes<Eigen::DenseIndex, 5> Index5;
327 typedef Eigen::DSizes<Eigen::DenseIndex, 2> Index2;
369 Eigen::DSizes<Eigen::DenseIndex, 5> indicesStart(1, 2, 3, 4, 5);
370 Eigen::DSizes<Eigen::DenseIndex, 5> indicesStop(2, 3, 4, 5, 6);
371 Eigen::DSizes<Eigen::DenseIndex, 5> strides(1, 1, 1, 1, 1);
419 typedef Eigen::DSizes<Eigen::DenseIndex, 2> Index2;
|
