ML functions
 
Loading...
Searching...
No Matches
velox::dl Namespace Reference

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.
 

Detailed Description

Namespace for deep learning-related utilities and kernels.

Enumeration Type Documentation

◆ KernelType

enum class velox::dl::KernelType
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.

Function Documentation

◆ kernelTypeToString()

std::string velox::dl::kernelTypeToString ( KernelType kernelType)

Converts a KernelType enum value to its string representation.

Parameters
kernelTypeThe kernel type to convert.
Returns
A string representing the kernel type.

◆ operator<<()

std::ostream & velox::dl::operator<< ( std::ostream & os,
KernelType kernelType )

Overloads the << operator for KernelType.

Parameters
osThe output stream.
kernelTypeThe kernel type to stream.
Returns
The output stream with the kernel type string representation.