/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/device_softmax.hpp Source File#
device_softmax.hpp
Go to the documentation of this file.
std::unique_ptr< DeviceSoftmax< InDataType, AccDataType, OutDataType, InElementwiseOp, AccElementwiseOp, Rank, NumReduceDim > > DeviceSoftmaxPtr
Definition: device_softmax.hpp:68
Definition: ck.hpp:264
Definition: device_base.hpp:76
Definition: device_softmax.hpp:24
virtual std::unique_ptr< BaseArgument > MakeArgumentPointer(const std::vector< index_t > inLengths, const std::vector< index_t > inStrides, const std::vector< int > reduceDims, double alpha, double beta, const void *in_dev, void *out_dev, InElementwiseOp in_elementwise_op, AccElementwiseOp acc_elementwise_op)=0
virtual std::unique_ptr< BaseInvoker > MakeInvokerPointer()=0