Home | History | Annotate | Download | only in Tensor

Lines Matching refs:NumInputDims

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)),
419 for (int i = 0; i < NumInputDims; ++i) {
424 eigen_assert(op.dims()[i] < NumInputDims);
447 if (NumInputDims > 0) {
448 array<Index, NumInputDims> input_strides;
451 for (int i = 1; i < NumInputDims; ++i) {
456 for (int i = NumInputDims - 2; i >= 0; --i) {
463 for (int i = 0; i < NumInputDims; ++i) {
519 reducing_inner_dims &= m_reduced[NumInputDims - 1 - i];
550 preserving_inner_dims &= m_reduced[NumInputDims - 1 - i];
739 array<bool, NumInputDims> m_reduced;