CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | List of all members
cutlass::multiply_add< Array< T, N >, Array< T, N >, Array< T, N > > Struct Template Reference

Fused multiply-add.

#include <functional.h>

Public Member Functions

CUTLASS_HOST_DEVICE Array< T, N > operator() (Array< T, N > const &a, Array< T, N > const &b, Array< T, N > const &c) const
 
CUTLASS_HOST_DEVICE Array< T, N > operator() (Array< T, N > const &a, T const &scalar, Array< T, N > const &c) const
 
CUTLASS_HOST_DEVICE Array< T, N > operator() (T const &scalar, Array< T, N > const &b, Array< T, N > const &c) const
 

Member Function Documentation

template<typename T , int N>
CUTLASS_HOST_DEVICE Array<T, N> cutlass::multiply_add< Array< T, N >, Array< T, N >, Array< T, N > >::operator() ( Array< T, N > const &  a,
Array< T, N > const &  b,
Array< T, N > const &  c 
) const
inline
template<typename T , int N>
CUTLASS_HOST_DEVICE Array<T, N> cutlass::multiply_add< Array< T, N >, Array< T, N >, Array< T, N > >::operator() ( Array< T, N > const &  a,
T const &  scalar,
Array< T, N > const &  c 
) const
inline
template<typename T , int N>
CUTLASS_HOST_DEVICE Array<T, N> cutlass::multiply_add< Array< T, N >, Array< T, N >, Array< T, N > >::operator() ( T const &  scalar,
Array< T, N > const &  b,
Array< T, N > const &  c 
) const
inline

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