| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_kernel_argmax_float_2d.cc | 36 Eigen::DSizes<Eigen::DenseIndex, 2> in_eig_sizes(in_sizes[0], in_sizes[1]);
40 Eigen::DSizes<Eigen::DenseIndex, 1> out_eig_sizes(out_sizes[0]);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| gather_nd_op_cpu_impl.h | 55 const Index loc, Eigen::array<Eigen::DenseIndex, IXDIM + 1>* ix) const {
67 operator()(const Eigen::array<Eigen::DenseIndex, 1>& loc_array) const {
69 Eigen::array<Eigen::DenseIndex, IXDIM + 1> ix;
70 Eigen::array<Eigen::DenseIndex, 2> ix_out;
105 const Eigen::DenseIndex batch_size = Tindices.dimension(0);
107 Eigen::Tensor<Eigen::DenseIndex, 1>::Dimensions reshape_dims{{ 1 }};
108 Eigen::array<Eigen::DenseIndex, 1> broadcast_dims{{ batch_size }};
111 Eigen::IndexList<Eigen::DenseIndex> broadcast_dims;
|
| strided_slice_op_impl.h | 89 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di;
90 Eigen::DSizes<Eigen::DenseIndex, NDIM> sizes_di;
100 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di;
101 Eigen::DSizes<Eigen::DenseIndex, NDIM> end_di;
102 Eigen::DSizes<Eigen::DenseIndex, NDIM> strides_di;
125 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di;
126 Eigen::DSizes<Eigen::DenseIndex, NDIM> end_di;
127 Eigen::DSizes<Eigen::DenseIndex, NDIM> strides_di;
148 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di;
149 Eigen::DSizes<Eigen::DenseIndex, NDIM> end_di
[all...] |
| matrix_diag_op.h | 37 operator()(const Eigen::array<Eigen::DenseIndex, 2>& coords) const {
38 Eigen::array<Eigen::DenseIndex, 3> diag_from_coords(
55 operator()(const Eigen::array<Eigen::DenseIndex, 3>& coords) const {
58 Eigen::array<Eigen::DenseIndex, 2> diag_coords({coords[0], coords[1]});
|
| betainc_op.h | 37 const typename Eigen::array<Eigen::DenseIndex, NDIM>& bcast_a,
39 const typename Eigen::array<Eigen::DenseIndex, NDIM>& bcast_b,
41 const typename Eigen::array<Eigen::DenseIndex, NDIM>& bcast_x,
|
| split_op.cc | 147 std::numeric_limits<Eigen::DenseIndex>::max()),
149 std::numeric_limits<Eigen::DenseIndex>::max()));
151 Eigen::DenseIndex prefix_dim_size;
152 Eigen::DenseIndex split_dim_size;
153 Eigen::DenseIndex suffix_dim_size;
156 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim);
164 Eigen::DSizes<Eigen::DenseIndex, 3> indices{0, 0, 0};
165 const Eigen::DSizes<Eigen::DenseIndex, 3> sizes{
186 Eigen::DSizes<Eigen::DenseIndex, 3> slice_indices;
187 Eigen::DSizes<Eigen::DenseIndex, 3> slice_sizes
[all...] |
| strided_slice_op.h | 34 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
35 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
36 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& strides) {
82 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
83 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
84 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& strides) {
108 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
109 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
110 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& strides) {
|
| slice_op.h | 31 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_indices,
32 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_sizes) {
|
| split_v_op.cc | 202 std::numeric_limits<Eigen::DenseIndex>::max()),
204 std::numeric_limits<Eigen::DenseIndex>::max()));
206 Eigen::DenseIndex prefix_dim_size;
207 Eigen::DenseIndex split_dim_size;
208 Eigen::DenseIndex suffix_dim_size;
211 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim);
215 Eigen::DSizes<Eigen::DenseIndex, 3> indices{0, 0, 0};
246 Eigen::DSizes<Eigen::DenseIndex, 3> sizes{
361 Eigen::DenseIndex prefix_dim_size;
362 Eigen::DenseIndex split_dim_size
[all...] |
| cwise_op_select.cc | 74 std::numeric_limits<Eigen::DenseIndex>::max()),
76 std::numeric_limits<Eigen::DenseIndex>::max()));
80 std::numeric_limits<Eigen::DenseIndex>::max()),
82 std::numeric_limits<Eigen::DenseIndex>::max()));
237 const Eigen::DenseIndex batch = cond_vec.size();
238 const Eigen::DenseIndex all_but_batch = then_flat_outer_dims.dimension(1);
241 Eigen::array<Eigen::DenseIndex, 2> broadcast_dims{{1, all_but_batch}};
242 Eigen::Tensor<Eigen::DenseIndex, 2>::Dimensions reshape_dims{{batch, 1}};
244 Eigen::IndexList<Eigen::type2index<1>, Eigen::DenseIndex> broadcast_dims;
246 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims
[all...] |
| data_format_ops.h | 44 Eigen::DSizes<Eigen::DenseIndex, 1> dimensions(
46 Eigen::DSizes<Eigen::DenseIndex, 1> result;
73 Eigen::DSizes<Eigen::DenseIndex, 1> dimensions(
75 Eigen::DSizes<Eigen::DenseIndex, 1> result;
|
| scatter_nd_op.h | 53 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix,
|
| reverse_sequence_op.h | 40 operator()(const Eigen::array<Eigen::DenseIndex, Dims>& coords) const {
41 Eigen::array<Eigen::DenseIndex, Dims> new_coords = coords;
|
| scatter_nd_op_cpu_impl.h | 94 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix,
103 const Eigen::DenseIndex batch_size = Tindices.dimension(0);
115 for (Eigen::DenseIndex loc = 0; loc < batch_size; ++loc) {
144 const Eigen::array<Eigen::DenseIndex, CPU_PROVIDED_IXDIM> \
176 const Eigen::array<Eigen::DenseIndex, IXDIM> output_shape_prefix,
185 const Eigen::DenseIndex batch_size = Tindices.dimension(0);
197 for (Eigen::DenseIndex loc = 0; loc < batch_size; ++loc) {
226 const Eigen::array<Eigen::DenseIndex, CPU_PROVIDED_IXDIM> \
|
| /external/eigen/unsupported/Eigen/src/NonLinearOptimization/ |
| r1mpyq.h | 8 void r1mpyq(DenseIndex m, DenseIndex n, Scalar *a, const std::vector<JacobiRotation<Scalar> > &v_givens, const std::vector<JacobiRotation<Scalar> > &w_givens)
10 typedef DenseIndex Index;
|
| fdjac1.h | 6 DenseIndex fdjac1(
11 DenseIndex ml, DenseIndex mu,
17 typedef DenseIndex Index;
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops.h | 36 inline Eigen::array<Eigen::DenseIndex, 2> x_offsets() const { return {0, 0}; }
38 inline Eigen::array<Eigen::DenseIndex, 2> x_extends() const {
42 inline Eigen::array<Eigen::DenseIndex, 2> h_offsets() const {
46 inline Eigen::array<Eigen::DenseIndex, 2> h_extends() const {
50 inline Eigen::array<Eigen::DenseIndex, 2> ru_r_offset() const {
54 inline Eigen::array<Eigen::DenseIndex, 2> ru_u_offset() const {
58 inline Eigen::array<Eigen::DenseIndex, 2> cell_extents() const {
94 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({batch_size_, 1});
95 Eigen::array<Eigen::DenseIndex, 2> b_ru_shape({1, b_ru.dimensions()[0]});
112 Eigen::array<Eigen::DenseIndex, 2> b_c_shape({1, b_c.dimensions()[0]})
[all...] |
| /external/eigen/unsupported/test/ |
| cxx11_tensor_generator.cpp | 17 float operator()(const array<Eigen::DenseIndex, 1>& coordinates) const {
37 float operator()(const array<Eigen::DenseIndex, 2>& coordinates) const {
67 internal::GaussianGenerator<float, Eigen::DenseIndex, 2> gaussian_gen(means, std_devs);
|
| splines.cpp | 216 for (Eigen::DenseIndex i=0; i<points.cols(); ++i)
228 for (Eigen::DenseIndex i=0; i<points.cols(); ++i)
242 const Eigen::DenseIndex numPoints = 100;
254 for (Eigen::DenseIndex i = 0; i < numPoints; ++i)
260 for (Eigen::DenseIndex i = 0; i < points.cols(); ++i)
|
| cxx11_tensor_index_list.cpp | 25 VERIFY_IS_EQUAL(static_cast<DenseIndex>(reduction_axis[0]), 0);
26 VERIFY_IS_EQUAL(static_cast<DenseIndex>(reduction_axis[1]), 1);
27 VERIFY_IS_EQUAL(static_cast<DenseIndex>(reduction_axis[2]), 2);
170 typedef Eigen::IndexPairList<Eigen::type2indexpair<0,10>, Eigen::IndexPair<DenseIndex>, Eigen::type2indexpair<2,12>> Dims2_b;
171 typedef Eigen::IndexPairList<Eigen::IndexPair<DenseIndex>, Eigen::type2indexpair<1,11>, Eigen::IndexPair<DenseIndex>> Dims2_c;
177 d2_b.set(1, Eigen::IndexPair<DenseIndex>(1,11));
180 d2_c.set(0, Eigen::IndexPair<DenseIndex>(Eigen::IndexPair<DenseIndex>(0,10)));
181 d2_c.set(1, Eigen::IndexPair<DenseIndex>(1,11)); // setting type2indexpair to correct value
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| multithreaded_conv.h | 70 Eigen::Tensor<float, 2, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned>
73 Eigen::Tensor<const float, 2, Eigen::RowMajor, Eigen::DenseIndex>,
78 Eigen::Tensor<float, 4, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned>
81 Eigen::Tensor<const float, 4, Eigen::RowMajor, Eigen::DenseIndex>,
93 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
130 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
131 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
143 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair;
144 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
|
| /external/tensorflow/tensorflow/core/util/ |
| bcast.h | 109 static Eigen::array<Eigen::DenseIndex, NDIMS> ToIndexArray(
112 Eigen::array<Eigen::DenseIndex, NDIMS> ret;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_fft_impl.h | 47 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> dims;
70 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims;
72 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims;
93 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices;
111 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims;
113 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims;
139 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> neg_target_indices;
142 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices;
143 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> neg_start_indices;
|
| /external/eigen/test/ |
| basicstuff.cpp | 218 Matrix<Scalar,2,1> m(raw), m2( (DenseIndex(raw[0])), (DenseIndex(raw[1])) );
219 Array<Scalar,2,1> a(raw), a2( (DenseIndex(raw[0])), (DenseIndex(raw[1])) );
224 for(int k=0; k<2; ++k) VERIFY(m2(k) == DenseIndex(raw[k]));
225 for(int k=0; k<2; ++k) VERIFY(a2(k) == DenseIndex(raw[k]));
229 m2( (DenseIndex(raw[0])), (DenseIndex(raw[1])) ),
232 Array<Scalar,1,2> a(raw), a2( (DenseIndex(raw[0])), (DenseIndex(raw[1])) )
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_slice.h | 123 Eigen::DSizes<Eigen::DenseIndex, NDIMS>* indices,
124 Eigen::DSizes<Eigen::DenseIndex, NDIMS>* sizes) const;
200 const TensorShape& shape, Eigen::DSizes<Eigen::DenseIndex, NDIMS>* indices,
201 Eigen::DSizes<Eigen::DenseIndex, NDIMS>* sizes) const {
|
