Home
last modified time | relevance | path

Searched refs:NumInputDims (Results 1 – 7 of 7) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorChipping.h132 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di…
133 static const int NumDims = NumInputDims-1;
155 EIGEN_STATIC_ASSERT((NumInputDims >= 1), YOU_MADE_A_PROGRAMMING_MISTAKE);
156 eigen_assert(NumInputDims > m_dim.actualDim());
162 for (int i = 0; i < NumInputDims; ++i) {
177 for (int i = NumInputDims-1; i > m_dim.actualDim(); --i) {
209 (static_cast<int>(Layout) == static_cast<int>(RowMajor) && m_dim.actualDim() == NumInputDims-1)) {
220 …static_cast<int>(Layout) == static_cast<int>(ColMajor) && m_dim.actualDim() == NumInputDims - 1) ||
250 m_dim.actualDim() == NumInputDims - 1)) {
253 m_dim.actualDim() == NumInputDims - 1) ||
[all …]
DTensorReduction.h61 const int NumInputDims = internal::array_size<InputDims>::value;
64 for (int i = 0; i < NumInputDims; ++i) {
80 const int NumInputDims = internal::array_size<InputDims>::value;
81 for (int i = 0; i < NumInputDims; ++i) {
388 static const int NumInputDims = internal::array_size<InputDimensions>::value;
390 static const int NumOutputDims = NumInputDims - NumReducedDims;
407 …static const bool ReducingInnerMostDims = internal::are_inner_most_dims<Dims, NumInputDims, Layout…
408 …static const bool PreservingInnerMostDims = internal::preserve_inner_most_dims<Dims, NumInputDims,…
414 EIGEN_STATIC_ASSERT((NumInputDims >= NumReducedDims), YOU_MADE_A_PROGRAMMING_MISTAKE);
415 …ATIC_ASSERT((!ReducingInnerMostDims | !PreservingInnerMostDims | (NumReducedDims == NumInputDims)),
[all …]
DTensorImagePatch.h155 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di…
156 static const int NumDims = NumInputDims + 1;
189 m_inputDepth = input_dims[NumInputDims-1];
190 m_inputRows = input_dims[NumInputDims-2];
191 m_inputCols = input_dims[NumInputDims-3];
271 m_dimensions[NumDims-1] = input_dims[NumInputDims-1];
DTensorVolumePatch.h170 …static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Di…
171 static const int NumDims = NumInputDims + 1;
203 m_inputDepth = input_dims[NumInputDims-1];
204 m_inputPlanes = input_dims[NumInputDims-2];
205 m_inputRows = input_dims[NumInputDims-3];
206 m_inputCols = input_dims[NumInputDims-4];
292 m_dimensions[NumDims-1] = input_dims[NumInputDims-1];
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h31 static constexpr int NumInputDims = internal::array_size< member
33 static constexpr int NumDims = NumInputDims + 1;
69 m_inputDepth = input_dims[NumInputDims - 1]; in CustomTensorEvaluator()
70 m_inputPlanes = input_dims[NumInputDims - 2]; in CustomTensorEvaluator()
71 m_inputRows = input_dims[NumInputDims - 3]; in CustomTensorEvaluator()
72 m_inputCols = input_dims[NumInputDims - 4]; in CustomTensorEvaluator()
182 m_dimensions[NumDims - 1] = input_dims[NumInputDims - 1]; in CustomTensorEvaluator()
Deigen_spatial_convolutions-inl.h615 const size_t NumInputDims = array_size< in computeBaseIndices() local
617 otherIndex = (NumInputDims == 3) ? 0 : patchIndex / m_fastNumPatches; in computeBaseIndices()
618 const Index patch2DIndex = (NumInputDims == 3) in computeBaseIndices()
Deigen_cuboid_convolution.h775 const size_t NumInputDims = array_size< in computeBaseIndices() local
779 otherIndex = (NumInputDims == 4) ? 0 : patchIndex / m_fastNumPatches; in computeBaseIndices()
782 const Index patch3DIndex = (NumInputDims == 4) in computeBaseIndices()