HomeSort by relevance Sort by last modified time
    Searched refs:NumInputDims (Results 1 - 6 of 6) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorChipping.h 132 static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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 } else if ((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...]
TensorReduction.h 61 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>::value;
408 static const bool PreservingInnerMostDims = internal::preserve_inner_most_dims<Dims, NumInputDims, Layout>::value;
414 EIGEN_STATIC_ASSERT((NumInputDims >= NumReducedDims), YOU_MADE_A_PROGRAMMING_MISTAKE);
415 EIGEN_STATIC_ASSERT((!ReducingInnerMostDims | !PreservingInnerMostDims | (NumReducedDims == NumInputDims)),
    [all...]
TensorImagePatch.h 155 static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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];
TensorVolumePatch.h 170 static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
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/
eigen_volume_patch.h 31 static const int NumInputDims = internal::array_size<
33 static const int NumDims = NumInputDims + 1;
68 m_inputDepth = input_dims[NumInputDims - 1];
69 m_inputPlanes = input_dims[NumInputDims - 2];
70 m_inputRows = input_dims[NumInputDims - 3];
71 m_inputCols = input_dims[NumInputDims - 4];
181 m_dimensions[NumDims - 1] = input_dims[NumInputDims - 1];
eigen_spatial_convolutions.h 404 const int NumInputDims = array_size<
406 otherIndex = (NumInputDims == 3) ? 0 : patchIndex / m_fastNumPatches;
407 const Index patch2DIndex = (NumInputDims == 3)
    [all...]

Completed in 291 milliseconds