/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/softmax/block/block_softmax_2d.hpp Source File#
block_softmax_2d.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
constexpr CK_TILE_HOST_DEVICE auto make_tuple(Xs &&... xs)
Definition: tuple.hpp:360
constexpr CK_TILE_HOST_DEVICE void sweep_tile(const F &f, UnpacksPerXDim={})
Definition: sweep_tile.hpp:231
CK_TILE_HOST_DEVICE_EXTERN BlockReduce2D(const T &, const typename T::DataType &) -> BlockReduce2D< T >
Definition: block_softmax_2d.hpp:21
remove_cvref_t< Problem_ > Problem
Definition: block_softmax_2d.hpp:22
CK_TILE_DEVICE void operator()(const DistributedTensor &x, DistributedTensor &y, number< dim >={})
Definition: block_softmax_2d.hpp:29
typename Problem::DataType DataType
Definition: block_softmax_2d.hpp:25
Definition: integral_constant.hpp:13
static constexpr CK_TILE_HOST_DEVICE T infinity()
Definition: numeric.hpp:38