Eigen::internal::DimInitializer< OutputDims > Struct Template Reference

#include <TensorReduction.h>

Static Public Member Functions
template<typename InputDims , typename ReducedDims >
static EIGEN_DEVICE_FUNC void	run (const InputDims &input_dims, const array< bool, internal::array_size< InputDims >::value > &reduced, OutputDims *output_dims, ReducedDims *reduced_dims)

Member Function Documentation

run()

template<typename OutputDims >

template<typename InputDims , typename ReducedDims >

static EIGEN_DEVICE_FUNC void Eigen::internal::DimInitializer< OutputDims >::run ( const InputDims &  input_dims, const array< bool, internal::array_size< InputDims >::value > &  reduced, OutputDims *  output_dims, ReducedDims *  reduced_dims  )

inlinestatic

                                                                                        {
    const int NumInputDims = internal::array_size<InputDims>::value;
    int outputIndex = 0;
    int reduceIndex = 0;
    for (int i = 0; i < NumInputDims; ++i) {
      if (reduced[i]) {
        (*reduced_dims)[reduceIndex] = input_dims[i];
        ++reduceIndex;
      } else {
        (*output_dims)[outputIndex] = input_dims[i];
        ++outputIndex;
      }
    }
  }

References i.

Referenced by Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::TensorReductionEvaluatorBase().

---

The documentation for this struct was generated from the following file:

• TensorReduction.h