/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/host/reference/reference_softmax.hpp Source File#
reference_softmax.hpp
Go to the documentation of this file.
__host__ constexpr __device__ auto rank([[maybe_unused]] const Layout< Shape, UnrolledDescriptorType > &layout)
Get layout rank (num elements in shape).
Definition: layout_utils.hpp:310
Definition: cluster_descriptor.hpp:13
CK_TILE_HOST auto make_ParallelTensorFunctor(F f, Xs... xs)
Definition: host_tensor.hpp:329
CK_TILE_HOST void reference_softmax(const HostTensor< InputType > &x, HostTensor< OutputType > &y, index_t dim=-1)
Definition: reference_softmax.hpp:14
Definition: host_tensor.hpp:336
std::size_t get_num_of_dimension() const
Definition: host_tensor.hpp:396
std::size_t get_length(std::size_t dim) const
Definition: host_tensor.hpp:388
std::size_t get_element_size() const
Definition: host_tensor.hpp:398
static constexpr CK_TILE_HOST_DEVICE T infinity()
Definition: numeric.hpp:38