Inheritance diagram for dlprim::SoftmaxBase:
Protected Member Functions | |
| bool | setup_kernel_params (int sm_range) |
Protected Attributes | |
| int | wg_size_ = 0 |
| int | items_per_wi_ = 0 |
| int | sm_range_ = -1 |
| int | nd_range_ |
The documentation for this class was generated from the following file:
- include/dlprim/ops/softmax.hpp
