DynamicUnaryOp Struct Reference

DynamicUnaryOp Struct Reference#

Composable Kernel: ck::tensor_operation::element_wise::DynamicUnaryOp Struct Reference
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
ck::tensor_operation::element_wise::DynamicUnaryOp Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions

__host__ __device__ DynamicUnaryOp ()=delete
__host__ __device__ DynamicUnaryOp (const Swish &swish)
__host__ __device__ DynamicUnaryOp (const Swish &&swish)
__host__ __device__ DynamicUnaryOp (const Sigmoid &)
__host__ __device__ DynamicUnaryOp (const Sigmoid &&)
__host__ __device__ DynamicUnaryOp (const PassThrough &)
__host__ __device__ DynamicUnaryOp (const PassThrough &&)
__host__ __device__ DynamicUnaryOp (const Logistic &logistic)
__host__ __device__ DynamicUnaryOp (const Logistic &&logistic)
__host__ __device__ DynamicUnaryOp (const TanH &)
__host__ __device__ DynamicUnaryOp (const TanH &&)
__host__ __device__ DynamicUnaryOp (const Relu &)
__host__ __device__ DynamicUnaryOp (const Relu &&)
__host__ __device__ DynamicUnaryOp (const SoftRelu &softrelu)
__host__ __device__ DynamicUnaryOp (const SoftRelu &&softrelu)
__host__ __device__ DynamicUnaryOp (const UnaryAbs &)
__host__ __device__ DynamicUnaryOp (const UnaryAbs &&)
__host__ __device__ DynamicUnaryOp (const Power &pow)
__host__ __device__ DynamicUnaryOp (const Power &&pow)
__host__ __device__ DynamicUnaryOp (const ClippedRelu &clippedrelu)
__host__ __device__ DynamicUnaryOp (const ClippedRelu &&clippedrelu)
__host__ __device__ DynamicUnaryOp (const LeakyRelu &leakyrelu)
__host__ __device__ DynamicUnaryOp (const LeakyRelu &&leakyrelu)
__host__ __device__ DynamicUnaryOp (const Elu &elu)
__host__ __device__ DynamicUnaryOp (const Elu &&elu)
__host__ __device__ DynamicUnaryOp (const DynamicUnaryOp &dynamic_op)=default
__host__ __device__ ~DynamicUnaryOp ()
template<typename Y, typename X>
__host__ __device__ void operator() (Y &y, const X &x) const
template<>
__host__ __device__ void operator() (bhalf_t &y, const bhalf_t &x) const

Public Attributes

UnaryOpType unary_op_type_
Swish swish_
Sigmoid sigmoid_
PassThrough pass_through_
Logistic logistic_
TanH tanh_
Relu relu_
SoftRelu soft_relu_
UnaryAbs unary_abs_
Power power_
ClippedRelu clipped_relu_
LeakyRelu leaky_relu_
Elu elu_

Static Public Attributes

static constexpr const char * name = "DynamicUnaryOp"

Constructor & Destructor Documentation

◆ DynamicUnaryOp() [1/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( )
delete

◆ DynamicUnaryOp() [2/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Swish & swish)
inline

◆ DynamicUnaryOp() [3/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Swish && swish)
inline

◆ DynamicUnaryOp() [4/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Sigmoid & )
inline

◆ DynamicUnaryOp() [5/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Sigmoid && )
inline

◆ DynamicUnaryOp() [6/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const PassThrough & )
inline

◆ DynamicUnaryOp() [7/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const PassThrough && )
inline

◆ DynamicUnaryOp() [8/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Logistic & logistic)
inline

◆ DynamicUnaryOp() [9/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Logistic && logistic)
inline

◆ DynamicUnaryOp() [10/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const TanH & )
inline

◆ DynamicUnaryOp() [11/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const TanH && )
inline

◆ DynamicUnaryOp() [12/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Relu & )
inline

◆ DynamicUnaryOp() [13/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Relu && )
inline

◆ DynamicUnaryOp() [14/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const SoftRelu & softrelu)
inline

◆ DynamicUnaryOp() [15/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const SoftRelu && softrelu)
inline

◆ DynamicUnaryOp() [16/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const UnaryAbs & )
inline

◆ DynamicUnaryOp() [17/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const UnaryAbs && )
inline

◆ DynamicUnaryOp() [18/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Power & pow)
inline

◆ DynamicUnaryOp() [19/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Power && pow)
inline

◆ DynamicUnaryOp() [20/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const ClippedRelu & clippedrelu)
inline

◆ DynamicUnaryOp() [21/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const ClippedRelu && clippedrelu)
inline

◆ DynamicUnaryOp() [22/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const LeakyRelu & leakyrelu)
inline

◆ DynamicUnaryOp() [23/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const LeakyRelu && leakyrelu)
inline

◆ DynamicUnaryOp() [24/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Elu & elu)
inline

◆ DynamicUnaryOp() [25/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const Elu && elu)
inline

◆ DynamicUnaryOp() [26/26]

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::DynamicUnaryOp ( const DynamicUnaryOp & dynamic_op)
default

◆ ~DynamicUnaryOp()

__host__ __device__ ck::tensor_operation::element_wise::DynamicUnaryOp::~DynamicUnaryOp ( )
inline

Member Function Documentation

◆ operator()() [1/2]

template<>
__host__ __device__ void ck::tensor_operation::element_wise::DynamicUnaryOp::operator() ( bhalf_t & y,
const bhalf_t & x ) const
inline

◆ operator()() [2/2]

template<typename Y, typename X>
__host__ __device__ void ck::tensor_operation::element_wise::DynamicUnaryOp::operator() ( Y & y,
const X & x ) const
inline

Member Data Documentation

◆ clipped_relu_

ClippedRelu ck::tensor_operation::element_wise::DynamicUnaryOp::clipped_relu_

◆ elu_

Elu ck::tensor_operation::element_wise::DynamicUnaryOp::elu_

◆ leaky_relu_

LeakyRelu ck::tensor_operation::element_wise::DynamicUnaryOp::leaky_relu_

◆ logistic_

Logistic ck::tensor_operation::element_wise::DynamicUnaryOp::logistic_

◆ name

const char* ck::tensor_operation::element_wise::DynamicUnaryOp::name = "DynamicUnaryOp"
staticconstexpr

◆ pass_through_

PassThrough ck::tensor_operation::element_wise::DynamicUnaryOp::pass_through_

◆ power_

Power ck::tensor_operation::element_wise::DynamicUnaryOp::power_

◆ relu_

Relu ck::tensor_operation::element_wise::DynamicUnaryOp::relu_

◆ sigmoid_

Sigmoid ck::tensor_operation::element_wise::DynamicUnaryOp::sigmoid_

◆ soft_relu_

SoftRelu ck::tensor_operation::element_wise::DynamicUnaryOp::soft_relu_

◆ swish_

Swish ck::tensor_operation::element_wise::DynamicUnaryOp::swish_

◆ tanh_

TanH ck::tensor_operation::element_wise::DynamicUnaryOp::tanh_

◆ unary_abs_

UnaryAbs ck::tensor_operation::element_wise::DynamicUnaryOp::unary_abs_

◆ unary_op_type_

UnaryOpType ck::tensor_operation::element_wise::DynamicUnaryOp::unary_op_type_
The documentation for this struct was generated from the following file: