Namespace for deep learning-related utilities and kernels. More...
Enumerations | |
| enum class | KernelType { MatMul , MatAdd , ReLU , Softmax , BatchNorm , Argmax , Sigmoid } |
| Enumeration of kernel types used in deep learning operations. More... | |
Functions | |
| std::string | kernelTypeToString (KernelType kernelType) |
| Converts a KernelType enum value to its string representation. | |
| std::ostream & | operator<< (std::ostream &os, KernelType kernelType) |
Overloads the << operator for KernelType. | |
Namespace for deep learning-related utilities and kernels.
|
strong |
Enumeration of kernel types used in deep learning operations.
| Enumerator | |
|---|---|
| MatMul | Matrix multiplication kernel. |
| MatAdd | Matrix addition kernel. |
| ReLU | Rectified Linear Unit activation kernel. |
| Softmax | Softmax activation kernel. |
| BatchNorm | Batch normalization kernel. |
| Argmax | Argmax operation kernel. |
| Sigmoid | Sigmoid activation kernel. |
| std::string velox::dl::kernelTypeToString | ( | KernelType | kernelType | ) |
Converts a KernelType enum value to its string representation.
| kernelType | The kernel type to convert. |
| std::ostream & velox::dl::operator<< | ( | std::ostream & | os, |
| KernelType | kernelType ) |
Overloads the << operator for KernelType.
| os | The output stream. |
| kernelType | The kernel type to stream. |