device_elementwise_normalization.hpp File Reference

device_elementwise_normalization.hpp File Reference#

Composable Kernel: device_elementwise_normalization.hpp File Reference
device_elementwise_normalization.hpp File Reference
#include <iostream>
#include <vector>
#include "ck/tensor_operation/gpu/device/device_base.hpp"

Go to the source code of this file.

Classes

struct  ck::tensor_operation::device::DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim >

Namespaces

namespace  ck
namespace  ck::tensor_operation
namespace  ck::tensor_operation::device

Typedefs

template<typename InDataTypeTuple, typename GammaDataType, typename BetaDataType, typename AccDataType, typename YDataType, typename XElementwiseOperation, typename YElementwiseOperation, index_t Rank, index_t NumReduceDim>
using ck::tensor_operation::device::DeviceElementwiseNormalizationPtr