#include <element_wise_operation.hpp>
|
| template<typename E, typename C, typename D0, typename D1> |
| __host__ __device__ constexpr void | operator() (E &e, const C &c, const D0 &d0, const D1 &d1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (ck::half_t &e, const float &c, const float &d0, const float &d1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (ck::bhalf_t &e, const float &c, const float &d0, const float &d1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (ck::half_t &e, const int &c, const ck::half_t &d0, const ck::half_t &d1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (ck::half_t &e, const int &c, const float &d0, const float &d1) const |
| template<> |
| __host__ __device__ constexpr void | operator() (ck::bhalf_t &e, const int &c, const float &d0, const float &d1) const |
|
| static constexpr const char * | name = "MultiplyMultiply" |
◆ operator()() [1/6]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyMultiply::operator() |
( |
ck::bhalf_t & | e, |
|
|
const float & | c, |
|
|
const float & | d0, |
|
|
const float & | d1 ) const |
|
inlineconstexpr |
◆ operator()() [2/6]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyMultiply::operator() |
( |
ck::bhalf_t & | e, |
|
|
const int & | c, |
|
|
const float & | d0, |
|
|
const float & | d1 ) const |
|
inlineconstexpr |
◆ operator()() [3/6]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyMultiply::operator() |
( |
ck::half_t & | e, |
|
|
const float & | c, |
|
|
const float & | d0, |
|
|
const float & | d1 ) const |
|
inlineconstexpr |
◆ operator()() [4/6]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyMultiply::operator() |
( |
ck::half_t & | e, |
|
|
const int & | c, |
|
|
const ck::half_t & | d0, |
|
|
const ck::half_t & | d1 ) const |
|
inlineconstexpr |
◆ operator()() [5/6]
template<>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyMultiply::operator() |
( |
ck::half_t & | e, |
|
|
const int & | c, |
|
|
const float & | d0, |
|
|
const float & | d1 ) const |
|
inlineconstexpr |
◆ operator()() [6/6]
template<typename E, typename C, typename D0, typename D1>
| __host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyMultiply::operator() |
( |
E & | e, |
|
|
const C & | c, |
|
|
const D0 & | d0, |
|
|
const D1 & | d1 ) const |
|
constexpr |
◆ name
| const char* ck::tensor_operation::element_wise::MultiplyMultiply::name = "MultiplyMultiply" |
|
staticconstexpr |
The documentation for this struct was generated from the following file: