Public Member Functions | List of all members
SqrtOutputKernel Struct Reference

Public Member Functions

template<typename Index , typename Scalar >
EIGEN_ALWAYS_INLINE void operator() (const internal::blas_data_mapper< Scalar, Index, ColMajor > &output_mapper, const TensorContractionParams &, Index, Index, Index num_rows, Index num_cols) const
 
template<typename Index , typename Scalar >
EIGEN_ALWAYS_INLINE void operator() (const internal::blas_data_mapper< Scalar, Index, ColMajor > &output_mapper, const TensorContractionParams &, Index, Index, Index num_rows, Index num_cols) const
 

Member Function Documentation

◆ operator()() [1/2]

template<typename Index , typename Scalar >
EIGEN_ALWAYS_INLINE void SqrtOutputKernel::operator() ( const internal::blas_data_mapper< Scalar, Index, ColMajor > &  output_mapper,
const TensorContractionParams &  ,
Index  ,
Index  ,
Index  num_rows,
Index  num_cols 
) const
inline
488  {
489  for (int i = 0; i < num_rows; ++i) {
490  for (int j = 0; j < num_cols; ++j) {
491  output_mapper(i, j) = std::sqrt(output_mapper(i, j));
492  }
493  }
494  }
sqrt
AnnoyingScalar sqrt(const AnnoyingScalar &x)
Definition: AnnoyingScalar.h:134
i
int i
Definition: BiCGSTAB_step_by_step.cpp:9
j
std::ptrdiff_t j
Definition: tut_arithmetic_redux_minmax.cpp:2

References i, j, and sqrt().

◆ operator()() [2/2]

template<typename Index , typename Scalar >
EIGEN_ALWAYS_INLINE void SqrtOutputKernel::operator() ( const internal::blas_data_mapper< Scalar, Index, ColMajor > &  output_mapper,
const TensorContractionParams &  ,
Index  ,
Index  ,
Index  num_rows,
Index  num_cols 
) const
inline
353  {
354  for (int i = 0; i < num_rows; ++i) {
355  for (int j = 0; j < num_cols; ++j) {
356  output_mapper(i, j) = std::sqrt(output_mapper(i, j));
357  }
358  }
359  }

References i, j, and sqrt().

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