Home
last modified time | relevance | path

Searched refs:m_dimensions (Results 1 – 24 of 24) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorMap.h60 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr) : m_data(dataPtr), m_dimensions() { in TensorMap()
67 …dex firstDimension, IndexTypes... otherDimensions) : m_data(dataPtr), m_dimensions(firstDimension,… in TensorMap()
73 …rMap(PointerArgType dataPtr, Index firstDimension) : m_data(dataPtr), m_dimensions(firstDimension)… in TensorMap()
78 …orMap(PointerArgType dataPtr, Index dim1, Index dim2) : m_data(dataPtr), m_dimensions(dim1, dim2) { in TensorMap()
82 …gType dataPtr, Index dim1, Index dim2, Index dim3) : m_data(dataPtr), m_dimensions(dim1, dim2, dim… in TensorMap()
86 …r, Index dim1, Index dim2, Index dim3, Index dim4) : m_data(dataPtr), m_dimensions(dim1, dim2, dim… in TensorMap()
90 …1, Index dim2, Index dim3, Index dim4, Index dim5) : m_data(dataPtr), m_dimensions(dim1, dim2, dim… in TensorMap()
96 : m_data(dataPtr), m_dimensions(dimensions) in TensorMap()
101 : m_data(dataPtr), m_dimensions(dimensions) in TensorMap()
105 : m_data(tensor.data()), m_dimensions(tensor.dimensions()) in TensorMap()
[all …]
DTensorStorage.h48 FixedDimensions m_dimensions; variable
61 EIGEN_STRONG_INLINE const FixedDimensions& dimensions() const { return m_dimensions; } in dimensions()
64 EIGEN_STRONG_INLINE DenseIndex size() const { return m_dimensions.TotalSize(); } in size()
77 EIGEN_DEVICE_FUNC TensorStorage() : m_data(0), m_dimensions() { in TensorStorage()
83 : m_data(0), m_dimensions(internal::template repeat<NumIndices_, Index>(0)) {} in TensorStorage()
85 …(internal::conditional_aligned_new_auto<T,(Options_&DontAlign)==0>(size)), m_dimensions(dimensions) in TensorStorage()
90 EIGEN_DEVICE_FUNC TensorStorage(DenseIndex... indices) : m_dimensions(indices...) { in TensorStorage()
91 …ernal::conditional_aligned_new_auto<T,(Options_&DontAlign)==0>(internal::array_prod(m_dimensions)); in TensorStorage()
96 …:conditional_aligned_new_auto<T,(Options_&DontAlign)==0>(internal::array_prod(other.m_dimensions))) in TensorStorage()
97 , m_dimensions(other.m_dimensions) in TensorStorage()
[all …]
DTensorImagePatch.h257 m_dimensions[0] = input_dims[0];
258 m_dimensions[1] = op.patch_rows();
259 m_dimensions[2] = op.patch_cols();
260 m_dimensions[3] = m_outputRows * m_outputCols;
262 m_dimensions[i] = input_dims[i-1];
271 m_dimensions[NumDims-1] = input_dims[NumInputDims-1];
272 m_dimensions[NumDims-2] = op.patch_rows();
273 m_dimensions[NumDims-3] = op.patch_cols();
274 m_dimensions[NumDims-4] = m_outputRows * m_outputCols;
276 m_dimensions[i] = input_dims[i];
[all …]
DTensorVolumePatch.h276 m_dimensions[0] = input_dims[0];
277 m_dimensions[1] = op.patch_planes();
278 m_dimensions[2] = op.patch_rows();
279 m_dimensions[3] = op.patch_cols();
280 m_dimensions[4] = m_outputPlanes * m_outputRows * m_outputCols;
282 m_dimensions[i] = input_dims[i-1];
292 m_dimensions[NumDims-1] = input_dims[NumInputDims-1];
293 m_dimensions[NumDims-2] = op.patch_planes();
294 m_dimensions[NumDims-3] = op.patch_rows();
295 m_dimensions[NumDims-4] = op.patch_cols();
[all …]
DTensorPadding.h112 m_dimensions = m_impl.dimensions();
114 m_dimensions[i] += m_padding[i].first + m_padding[i].second;
122 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
124 m_outputStrides[NumDims] = m_outputStrides[NumDims-1] * m_dimensions[NumDims-1];
130 m_outputStrides[i+1] = m_outputStrides[i+2] * m_dimensions[i+1];
132 m_outputStrides[0] = m_outputStrides[1] * m_dimensions[0];
136 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
210 index >= m_dimensions[dim_index] - m_padding[dim_index].second);
213 (index >= m_dimensions[dim_index] - m_padding[dim_index].second);
269 const Index firstPaddedRight = (m_dimensions[i] - m_padding[i].second) * m_outputStrides[i];
[all …]
DTensorReverse.h128 m_dimensions = m_impl.dimensions();
132 m_strides[i] = m_strides[i-1] * m_dimensions[i-1];
137 m_strides[i] = m_strides[i+1] * m_dimensions[i+1];
143 const Dimensions& dimensions() const { return m_dimensions; }
162 idx = m_dimensions[i] - idx - 1;
167 inputIndex += (m_dimensions[0] - index - 1);
176 idx = m_dimensions[i] - idx - 1;
181 inputIndex += (m_dimensions[NumDims-1] - index - 1);
228 Dimensions m_dimensions;
264 const Dimensions& dimensions() const { return this->m_dimensions; }
DTensorInflation.h101 m_dimensions = m_impl.dimensions();
104 m_dimensions[i] = (m_dimensions[i] - 1) * op.strides()[i] + 1;
117 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
124 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
130 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
208 const double output_size = m_dimensions.TotalSize();
219 Dimensions m_dimensions;
DTensorPatch.h109 m_dimensions[i] = patch_dims[i];
112 m_dimensions[NumDims-1] = num_patches;
122 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
126 m_dimensions[i+1] = patch_dims[i];
129 m_dimensions[0] = num_patches;
139 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
144 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
259 Dimensions m_dimensions;
DTensorCustomOp.h103 m_dimensions = op.func().dimensions(op.expression());
106 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
146 data, m_dimensions);
150 Dimensions m_dimensions;
259 m_dimensions = op.func().dimensions(op.lhsExpression(), op.rhsExpression());
262 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
300 TensorMap<Tensor<Scalar, NumDims, Layout> > result(data, m_dimensions);
304 Dimensions m_dimensions;
DTensorGenerator.h103 m_dimensions = impl.dimensions();
108 m_strides[i] = m_strides[i - 1] * m_dimensions[i - 1];
113 m_strides[i] = m_strides[i + 1] * m_dimensions[i + 1];
118 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
178 Dimensions m_dimensions;
DTensorSyclExtractFunctors.h38 const Dimensions m_dimensions; member
39 const Dimensions& dimensions() const { return m_dimensions; } in dimensions()
41 : m_dimensions(expr.dimensions()) {} in FunctorExtractor()
158 const Dimensions m_dimensions;
159 const Dimensions& dimensions() const { return m_dimensions; }
161 : m_dimensions(DimConstr<Dimensions, Evaluator::NumOutputDims>::getDim(expr.dimensions())) {}
DTensorStriding.h122 m_dimensions = m_impl.dimensions();
124 m_dimensions[i] = ceilf(static_cast<float>(m_dimensions[i]) / op.strides()[i]);
132 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
141 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
149 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
249 Dimensions m_dimensions;
DTensorMorphing.h113 : m_impl(op.expression(), device), m_dimensions(op.dimensions())
125 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
155 NewDimensions m_dimensions;
322 …: m_impl(op.expression(), device), m_device(device), m_dimensions(op.sizes()), m_offsets(op.startI…
363 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
467 if (m_dimensions[i] != m_impl.dimensions()[i]) {
470 if (m_dimensions[j] > 1) {
480 if (m_dimensions[i] != m_impl.dimensions()[i]) {
483 if (m_dimensions[j] > 1) {
524 Dimensions m_dimensions;
[all …]
DTensorShuffling.h125 m_dimensions[i] = input_dims[shuffle[i]];
135 m_outputStrides[i] = m_outputStrides[i - 1] * m_dimensions[i - 1];
142 m_outputStrides[i] = m_outputStrides[i + 1] * m_dimensions[i + 1];
151 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
210 Dimensions m_dimensions;
DTensorConcatenation.h142 m_dimensions[i] = lhs_dims[i];
146 m_dimensions[i] = lhs_dims[i] + rhs_dims[i];
150 m_dimensions[i] = lhs_dims[i];
162 m_outputStrides[j] = m_outputStrides[j-1] * m_dimensions[j-1];
172 m_outputStrides[j] = m_outputStrides[j+1] * m_dimensions[j+1];
177 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
281 Dimensions m_dimensions;
DTensorFFT.h147 m_dimensions[i] = input_dims[i];
153 m_strides[i] = m_strides[i - 1] * m_dimensions[i - 1];
158 m_strides[i] = m_strides[i + 1] * m_dimensions[i + 1];
161 m_size = m_dimensions.TotalSize();
165 return m_dimensions;
218 Index line_len = m_dimensions[dim];
538 const Index partial_m_stride = m_strides[i] / m_dimensions[omitted_dim];
547 const Index partial_m_stride = m_strides[i] / m_dimensions[omitted_dim];
566 Dimensions m_dimensions;
DTensorBroadcasting.h126 m_dimensions[i] = input_dims[i] * broadcast[i];
134 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];
141 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];
146 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
383 Dimensions m_dimensions;
DTensorLayoutSwap.h130 m_dimensions[i] = m_impl.dimensions()[NumDims-1-i];
138 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
168 Dimensions m_dimensions;
DTensorReduction.h429 internal::DimInitializer<Dimensions>::run(input_dims, m_reduced, &m_dimensions, &m_reducedDims);
436 m_outputStrides[i] = m_outputStrides[i - 1] * m_dimensions[i - 1];
441 m_outputStrides[i] = m_outputStrides[i + 1] * m_dimensions[i + 1];
480 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
502 const Index num_coeffs_to_preserve = internal::array_prod(m_dimensions);
525 const Index num_coeffs_to_preserve = internal::array_prod(m_dimensions);
558 const Index num_coeffs_to_preserve = internal::array_prod(m_dimensions);
634 … if (((firstIndex % m_dimensions[innermost_dim]) + PacketSize - 1) < m_dimensions[innermost_dim]) {
741 Dimensions m_dimensions;
DTensorConvolution.h333 m_dimensions = m_inputImpl.dimensions();
340 m_dimensions[index] = result_dim;
351 m_outputStride[i] = m_outputStride[i - 1] * m_dimensions[i - 1];
359 m_dimensions[index] = result_dim;
370 m_outputStride[i] = m_outputStride[i + 1] * m_dimensions[i + 1];
375 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
542 Dimensions m_dimensions;
780 m_dimensions = m_inputImpl.dimensions();
786 m_dimensions[index] = result_dim;
795 EIGEN_DEVICE_FUNC const Dimensions& dimensions() const { return m_dimensions; }
[all …]
DTensorContraction.h249 m_dimensions[dim_idx] = eval_left_dims[i];
276 m_dimensions[dim_idx] = eval_right_dims[i];
323 numext::swap(m_dimensions[i], m_dimensions[j]);
328 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
544 Dimensions m_dimensions;
DTensorChipping.h164 m_dimensions[j] = input_dims[i];
186 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
300 Dimensions m_dimensions;
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h166 m_dimensions[0] = input_dims[0]; in CustomTensorEvaluator()
167 m_dimensions[1] = op.patch_planes(); in CustomTensorEvaluator()
168 m_dimensions[2] = op.patch_rows(); in CustomTensorEvaluator()
169 m_dimensions[3] = op.patch_cols(); in CustomTensorEvaluator()
170 m_dimensions[4] = m_outputPlanes * m_outputRows * m_outputCols; in CustomTensorEvaluator()
172 m_dimensions[i] = input_dims[i - 1]; in CustomTensorEvaluator()
182 m_dimensions[NumDims - 1] = input_dims[NumInputDims - 1]; in CustomTensorEvaluator()
183 m_dimensions[NumDims - 2] = op.patch_planes(); in CustomTensorEvaluator()
184 m_dimensions[NumDims - 3] = op.patch_rows(); in CustomTensorEvaluator()
185 m_dimensions[NumDims - 4] = op.patch_cols(); in CustomTensorEvaluator()
[all …]
/external/deqp/external/vulkancts/modules/vulkan/texture/
DvktTextureFilteringExplicitLodTests.cpp1017 , m_dimensions (dimensions) in Texture2DGradientTestCase()
1060 const deUint8 numLevels = (deUint8) (1 + deLog2Floor32(de::max(m_dimensions[0], in genTestCaseData()
1061 m_dimensions[1]))); in genTestCaseData()
1067 m_dimensions, in genTestCaseData()
1087 const IVec3 m_dimensions; member in vkt::texture::__anonfaaa72040111::Texture2DGradientTestCase
1108 m_testCase->m_dimensions[0], in generate()
1109 m_testCase->m_dimensions[1])); in generate()
1111 const deUint8 numLevels = (deUint8) (1 + deLog2Floor32(de::max(m_testCase->m_dimensions[0], in generate()
1112 m_testCase->m_dimensions[1]))); in generate()
1165 for (deInt32 i = 0; i < 2 * m_testCase->m_dimensions[0] + 1; ++i) in getSampleArgs()
[all …]