colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

doc: fix some comments (#322)

This commit is contained in:
Dahan Gong 2022-03-16 23:21:20 -05:00 committed by GitHub
parent 2a8936dfed
commit aaaeda1846
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/tim/vx/ops/activations.h
View File

@ -51,12 +51,12 @@ namespace ops {
*
* HardSwish(x) : 0 if x <= -3; x(x + 3)/6 if -3 < x < 3; x if x >= 3
*
* Mish(x) : x if x >= 0 else alpha * x
*
* HardSigmoid(x) : min(max(alpha*x + beta, 0), 1)
*
* SoftRelu(x) : log(1 + e^x). Also known as SoftPlus.
*
* Mish(x) : x * tanh(softrelu(x))
*
* LeakyRelu(x) : alpha * x if x <= 0; x if x > 0. alpha is a scalar.
*
* Prelu(x) : alpha * x if x <= 0; x if x > 0. alpha is a tensor.

2
include/tim/vx/ops/localresponsenormalization.h
View File

@ -33,7 +33,7 @@
* ```
* sqr_sum[a, b, c, d] = sum(
* pow(input[a, b, c, d - depth_radius : d + depth_radius + 1], 2))
* output = input / pow((bias + alpha * sqr_sum), beta)
* output = input / pow((bias + alpha * sqr_sum), beta)
* ```
*/