Home
last modified time | relevance | path

Searched refs:DSizes (Results 1 – 25 of 114) sorted by relevance

12345

/external/eigen/unsupported/test/
Dcxx11_tensor_morphing.cpp100 Eigen::DSizes<ptrdiff_t, 5> indices(1,2,3,4,5); in test_simple_slice()
101 Eigen::DSizes<ptrdiff_t, 5> sizes(1,1,1,1,1); in test_simple_slice()
106 Eigen::DSizes<ptrdiff_t, 5> indices2(1,1,3,4,5); in test_simple_slice()
107 Eigen::DSizes<ptrdiff_t, 5> sizes2(1,1,2,2,3); in test_simple_slice()
123 DSizes<DenseIndex, 1> offsets; in test_const_slice()
145 Eigen::DSizes<ptrdiff_t, 2> indices1(1,2); in test_slice_in_expr()
146 Eigen::DSizes<ptrdiff_t, 2> sizes1(3,3); in test_slice_in_expr()
147 Eigen::DSizes<ptrdiff_t, 2> indices2(0,2); in test_slice_in_expr()
148 Eigen::DSizes<ptrdiff_t, 2> sizes2(3,1); in test_slice_in_expr()
160 …1, DataLayout> tensor6 = tensor4.reshape(DSizes<ptrdiff_t, 1>(7*7)).exp().slice(DSizes<ptrdiff_t, … in test_slice_in_expr()
[all …]
Dcxx11_tensor_dimension.cpp19 Eigen::DSizes<int, 3> dimensions(2,3,7); in test_dynamic_size()
42 Eigen::DSizes<unsigned int, 3> dyn((unsigned int)2,(unsigned int)3,(unsigned int)7); in test_match()
46 Eigen::DSizes<int, 3> dyn1(2,3,7); in test_match()
47 Eigen::DSizes<int, 2> dyn2(2,3); in test_match()
58 Eigen::DSizes<ptrdiff_t, 0> dscalar; in test_rank_zero()
Dcxx11_tensor_custom_op.cpp18 DSizes<DenseIndex, 2> dimensions(const Tensor<float, 2>& input) const { in dimensions()
19 DSizes<DenseIndex, 2> result; in dimensions()
33 Eigen::DSizes<DenseIndex, 2> offsets(1,1); in eval()
34 Eigen::DSizes<DenseIndex, 2> extents(output.dimension(0)-1, output.dimension(1)-1); in eval()
62 …DSizes<DenseIndex, 3> dimensions(const Tensor<float, 3>& input1, const Tensor<float, 3>& input2) c… in dimensions()
63 DSizes<DenseIndex, 3> result; in dimensions()
Dcxx11_tensor_custom_index.cpp27 using NormalIndex = DSizes<ptrdiff_t, 4>; in test_map_as_index()
49 using NormalIndex = DSizes<ptrdiff_t, 4>; in test_matrix_as_index()
67 DSizes<ptrdiff_t, 4> coeff(1,2,4,1); in test_varlist_as_index()
82 DSizes<ptrdiff_t, 4> coeff(1,2,4,1); in test_sizes_as_index()
Dcxx11_tensor_ref.cpp96 Eigen::DSizes<ptrdiff_t, 5> indices(1,2,3,4,5); in test_slice()
97 Eigen::DSizes<ptrdiff_t, 5> sizes(1,1,1,1,1); in test_slice()
101 Eigen::DSizes<ptrdiff_t, 5> indices2(1,1,3,4,5); in test_slice()
102 Eigen::DSizes<ptrdiff_t, 5> sizes2(1,1,2,2,3); in test_slice()
112 Eigen::DSizes<ptrdiff_t, 5> indices3(0,0,0,0,0); in test_slice()
113 Eigen::DSizes<ptrdiff_t, 5> sizes3(2,3,1,1,1); in test_slice()
209 DSizes<Eigen::DenseIndex, 4> shuffle_dims(0, 1, 2, 3); in test_nested_ops_with_ref()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_impl.h89 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di; in HandleStridedSliceCase()
90 Eigen::DSizes<Eigen::DenseIndex, NDIM> sizes_di; in HandleStridedSliceCase()
100 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di; in HandleStridedSliceCase()
101 Eigen::DSizes<Eigen::DenseIndex, NDIM> end_di; in HandleStridedSliceCase()
102 Eigen::DSizes<Eigen::DenseIndex, NDIM> strides_di; in HandleStridedSliceCase()
125 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di; in HandleStridedSliceGradCase()
126 Eigen::DSizes<Eigen::DenseIndex, NDIM> end_di; in HandleStridedSliceGradCase()
127 Eigen::DSizes<Eigen::DenseIndex, NDIM> strides_di; in HandleStridedSliceGradCase()
148 Eigen::DSizes<Eigen::DenseIndex, NDIM> begin_di; in operator()
149 Eigen::DSizes<Eigen::DenseIndex, NDIM> end_di; in operator()
[all …]
Dsplit_lib.h30 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_indices,
31 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_sizes);
38 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices,
39 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes);
47 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices,
48 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes);
57 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices,
58 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes);
Dstrided_slice_op.h34 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices, in operator()
35 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices, in operator()
36 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& strides) { in operator()
40 Eigen::DSizes<int, NDIMS> start_i, stop_i, strides_i; in operator()
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) {
89 Eigen::DSizes<int, NDIMS> start_i, stop_i, strides_i;
108 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& start_indices,
109 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& stop_indices,
[all …]
Deigen_backward_spatial_convolutions.h61 const DSizes<typename internal::traits<OutputBackward>::Index,
67 const DSizes<typename internal::traits<OutputBackward>::Index,
75 const DSizes<typename internal::traits<OutputBackward>::Index,
80 const DSizes<typename internal::traits<OutputBackward>::Index,
86 const DSizes<typename internal::traits<OutputBackward>::Index,
91 const DSizes<typename internal::traits<OutputBackward>::Index,
212 DSizes<TensorIndex, 2> kernel_dims;
226 DSizes<TensorIndex, 2> pre_contract_dims;
254 DSizes<TensorIndex, NumDims> post_contract_dims;
329 const DSizes<typename internal::traits<Input>::Index, 4>,
[all …]
Dtile_ops_impl.h31 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, in operator()
32 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes, in operator()
46 const Eigen::DSizes<Eigen::DenseIndex, 0>&,
47 const Eigen::DSizes<Eigen::DenseIndex, 0>&,
62 const Eigen::DSizes<Eigen::DenseIndex, REDUCEDNDIM>& reduce_dim,
63 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& reshape_dim) const {
Deigen_pooling.h42 const Eigen::DSizes<typename internal::traits<Input>::Index,
52 const Eigen::DSizes<typename internal::traits<Input>::Index,
92 Eigen::DSizes<TensorIndex, internal::traits<Input>::NumDimensions>
161 const Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions>,
165 const Eigen::DSizes<DenseIndex, 3>,
171 const Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions>,
176 const Eigen::DSizes<DenseIndex, 3>,
205 Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions> in CuboidMaxPooling()
228 Eigen::DSizes<DenseIndex, 3> pre_reduce_dims; in CuboidMaxPooling()
399 const Eigen::DSizes<typename internal::traits<Input>::Index,
[all …]
Deigen_backward_cuboid_convolutions.h50 const DSizes<typename internal::traits<OutputBackward>::Index,
56 const DSizes<typename internal::traits<OutputBackward>::Index,
64 const DSizes<typename internal::traits<OutputBackward>::Index,
69 const DSizes<typename internal::traits<OutputBackward>::Index,
75 const DSizes<typename internal::traits<OutputBackward>::Index,
80 const DSizes<typename internal::traits<OutputBackward>::Index,
210 DSizes<TensorIndex, 2> kernel_dims;
225 DSizes<TensorIndex, 2> pre_contract_dims;
255 DSizes<TensorIndex, NumDims> post_contract_dims;
334 const Eigen::DSizes<typename internal::traits<Input>::Index,
[all …]
Ddata_format_ops.h54 VecPermute(const Eigen::DSizes<Eigen::DenseIndex, 8>& dst) : dst_(dst) {} in VecPermute()
55 Eigen::DSizes<Eigen::DenseIndex, 1> dimensions( in dimensions()
57 Eigen::DSizes<Eigen::DenseIndex, 1> result; in dimensions()
70 Eigen::DSizes<Eigen::DenseIndex, 8> dst_;
78 const Eigen::DSizes<Eigen::DenseIndex, 8>& dst) { in operator()
Dslice_op.h31 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_indices, in operator()
32 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_sizes) { in operator()
36 Eigen::DSizes<int, NDIMS> indices; in operator()
40 Eigen::DSizes<int, NDIMS> sizes; in operator()
Dbatch_norm_op.h38 Eigen::DSizes<int, 2> rest_by_depth(rest_size, depth); in operator()
40 Eigen::DSizes<int, 2> rest_by_one(rest_size, 1); in operator()
41 Eigen::DSizes<int, 2> one_by_depth(1, depth); in operator()
42 Eigen::DSizes<int, 2> depth_by_one(depth, 1); in operator()
90 Eigen::DSizes<Index, 2> rest_by_depth(rest_size, depth); in operator()
92 Eigen::DSizes<Index, 2> rest_by_one(rest_size, 1); in operator()
93 Eigen::DSizes<Index, 2> one_by_depth(1, depth); in operator()
Dbias_op.h36 Eigen::DSizes<int64_t, 1> one_d(input.size()); in operator()
37 Eigen::DSizes<int64_t, 1> bcast(rest_size); in operator()
43 Eigen::DSizes<int, 1> one_d(input.size()); in operator()
44 Eigen::DSizes<int, 1> bcast(rest_size); in operator()
Dconv_2d.h35 const Eigen::DSizes<IndexType, Dims>& order, in operator()
46 const Eigen::DSizes<IndexType, Dims>& strides, in operator()
48 const Eigen::DSizes<IndexType, Dims>& order, in operator()
157 Eigen::DSizes<IndexType, 3> merged_dims;
170 Eigen::DSizes<IndexType, 3> shuffling_perm =
171 Eigen::DSizes<IndexType, 3>(2, 1, 0);
173 Eigen::DSizes<IndexType, NDIMS> expanded_dims;
196 const Eigen::DSizes<IndexType, 4>& shuffle,
198 Eigen::DSizes<IndexType, 3> merged_dims;
199 Eigen::DSizes<IndexType, 4> expanded_dims;
[all …]
Dsplit_lib_cpu.cc31 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, in operator ()()
32 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes) { in operator ()()
52 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, in DEFINE_CPU_KERNELS()
53 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes) { in DEFINE_CPU_KERNELS()
Deigen_attention.h68 DSizes<Index, 4> dimensions(const Input& input) const { in dimensions()
75 DSizes<Index, 4> dims = in.dimensions(); in dimensions()
125 DSizes<Index, 3> slice_offset(0, offset_x, offset_y); in eval()
126 DSizes<Index, 3> slice_extent(num_channels, width_, height_); in eval()
127 DSizes<Index, 3> base_offset(0, 0, 0); in eval()
171 DSizes<Index, 3> glimpse_size(num_channels, width_, height_); in eval()
183 DSizes<Index, 2> glimpse_size(width_, height_); in eval()
184 DSizes<Index, 2> input_size(input_width, input_height); in eval()
Dslice_op.cc218 Eigen::DSizes<Eigen::DenseIndex, NDIM> indices; in HandleCase()
219 Eigen::DSizes<Eigen::DenseIndex, NDIM> sizes; in HandleCase()
239 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, \
240 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes); \
278 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, \
279 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes); \
344 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& indices, \
345 const Eigen::DSizes<Eigen::DenseIndex, NDIM>& sizes); \
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h70 const DSizes<typename internal::traits<Input>::Index,
76 const DSizes<typename internal::traits<Input>::Index, 2>,
79 const DSizes<typename internal::traits<Input>::Index, 2>,
83 const DSizes<typename internal::traits<Input>::Index,
89 const DSizes<typename internal::traits<Input>::Index, 2>,
92 const DSizes<typename internal::traits<Input>::Index, 2>,
166 DSizes<TensorIndex, 2> pre_contract_dims;
187 DSizes<TensorIndex, NumDims> post_contract_dims;
204 DSizes<TensorIndex, 2> kernel_dims;
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h73 const DSizes<typename internal::traits<Input>::Index,
79 const DSizes<typename internal::traits<Input>::Index, 2>,
82 const DSizes<typename internal::traits<Input>::Index, 2>,
86 const DSizes<typename internal::traits<Input>::Index,
92 const DSizes<typename internal::traits<Input>::Index, 2>,
95 const DSizes<typename internal::traits<Input>::Index, 2>,
169 DSizes<TensorIndex, 2> pre_contract_dims;
190 DSizes<TensorIndex, NumDims> post_contract_dims;
207 DSizes<TensorIndex, 2> kernel_dims;
/external/tensorflow/tensorflow/core/framework/
Dtensor_types.h107 template <typename DSizes>
108 Eigen::DSizes<Index32, DSizes::count> To32BitDims(const DSizes& in) { in To32BitDims()
109 Eigen::DSizes<Index32, DSizes::count> out; in To32BitDims()
110 for (int i = 0; i < DSizes::count; ++i) { in To32BitDims()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h45 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> dims; in EigenFftC2C()
68 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftR2C()
70 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; in EigenFftR2C()
89 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices; in EigenFftR2C()
107 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftC2R()
109 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; in EigenFftC2R()
132 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> neg_target_indices; in EigenFftC2R()
135 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices; in EigenFftC2R()
136 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> neg_start_indices; in EigenFftC2R()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDimensions.h261 struct DSizes : array<DenseIndex, NumDims> {
273 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DSizes() {
278 EIGEN_DEVICE_FUNC explicit DSizes(const array<DenseIndex, NumDims>& a) : Base(a) { }
280 EIGEN_DEVICE_FUNC explicit DSizes(const DenseIndex i0) {
287 …EIGEN_STRONG_INLINE explicit DSizes(DenseIndex firstDimension, DenseIndex secondDimension, IndexTy…
291 EIGEN_DEVICE_FUNC DSizes(const DenseIndex i0, const DenseIndex i1) {
296 EIGEN_DEVICE_FUNC DSizes(const DenseIndex i0, const DenseIndex i1, const DenseIndex i2) {
302 …EIGEN_DEVICE_FUNC DSizes(const DenseIndex i0, const DenseIndex i1, const DenseIndex i2, const Dens…
309 …EIGEN_DEVICE_FUNC DSizes(const DenseIndex i0, const DenseIndex i1, const DenseIndex i2, const Dens…
319 EIGEN_DEVICE_FUNC DSizes& operator = (const array<DenseIndex, NumDims>& other) {
[all …]

12345