gpu::xetla::subgroup::dropout_op_t< dtype_mask, arch_tag, enable > Struct Template Reference

Is the dropout op functor. More...

#include <tile_op_functor.hpp>

Detailed Description

template<typename dtype_mask, gpu_arch arch_tag, class enable = void>
struct gpu::xetla::subgroup::dropout_op_t< dtype_mask, arch_tag, enable >

Is the dropout op functor.

Load the mask from memory and get input from matAcc, do the scaling and zero out, update the output in place. Used in epilogue::tile_op or chained_tile_op.

Template Parameters

dtype_mask	Is the mask data type.
arch_tag	Is the hardware architecture tag.