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

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReduction.h 389 static const int NumReducedDims = internal::array_size<Dims>::value;
390 static const int NumOutputDims = NumInputDims - NumReducedDims;
414 EIGEN_STATIC_ASSERT((NumInputDims >= NumReducedDims), YOU_MADE_A_PROGRAMMING_MISTAKE);
415 EIGEN_STATIC_ASSERT((!ReducingInnerMostDims | !PreservingInnerMostDims | (NumReducedDims == NumInputDims)),
422 for (int i = 0; i < NumReducedDims; ++i) {
515 for (int i = 0; i < NumReducedDims; ++i) {
548 for (int i = 0; i < NumReducedDims; ++i) {
604 internal::GenericDimReducer<NumReducedDims-1, Self, Op>::reduce(*this, firstInput(index), reducer, &accum);
637 internal::InnerMostDimPreserver<NumReducedDims-1, Self, Op>::reduce(*this, firstIndex, reducer, &accum);
    [all...]
TensorReductionSycl.h 228 GenericDimReducer<DeiceSelf::NumReducedDims-1, DeiceSelf, Op>::reduce(device_self_evaluator, device_self_evaluator.firstInput(globalid),const_cast<Op&>(functor), &accum);

Completed in 401 milliseconds