ScaleAndResetNaNToMinusInfinity Struct Reference

ScaleAndResetNaNToMinusInfinity Struct Reference#

Composable Kernel: ck::tensor_operation::element_wise::ScaleAndResetNaNToMinusInfinity Struct Reference
ck::tensor_operation::element_wise::ScaleAndResetNaNToMinusInfinity Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions

__host__ __device__ ScaleAndResetNaNToMinusInfinity (float scale)
template<typename Y, typename X>
__host__ __device__ void operator() (Y &y, const X &x) const
template<>
__host__ __device__ void operator() (float &y, const float &x) const

Public Attributes

float scale_

Static Public Attributes

static constexpr const char * name = "ScaleAndResetNaNToMinusInfinity"

Constructor & Destructor Documentation

◆ ScaleAndResetNaNToMinusInfinity()

__host__ __device__ ck::tensor_operation::element_wise::ScaleAndResetNaNToMinusInfinity::ScaleAndResetNaNToMinusInfinity ( float scale)
inline

Member Function Documentation

◆ operator()() [1/2]

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

◆ operator()() [2/2]

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

Member Data Documentation

◆ name

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

◆ scale_

float ck::tensor_operation::element_wise::ScaleAndResetNaNToMinusInfinity::scale_

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