ROCm / composable_kernel

Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
https://rocm.docs.amd.com/projects/composable_kernel/en/latest/
Other
297 stars 113 forks source link

Adding Missed Activation Functions for Grouped 2D/3D Convolutions #1323

Closed ThruptiRajLakshmanaGowda closed 3 months ago

ThruptiRajLakshmanaGowda commented 3 months ago

Added Swish, PassThrough and Logistic function.