From aaaeda1846d6a9467aea01eecf365b5e58f22464 Mon Sep 17 00:00:00 2001 From: Dahan Gong Date: Wed, 16 Mar 2022 23:21:20 -0500 Subject: [PATCH] doc: fix some comments (#322) --- include/tim/vx/ops/activations.h | 4 ++-- include/tim/vx/ops/localresponsenormalization.h | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/include/tim/vx/ops/activations.h b/include/tim/vx/ops/activations.h index 0f2070d..e8f3fd4 100644 --- a/include/tim/vx/ops/activations.h +++ b/include/tim/vx/ops/activations.h @@ -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. diff --git a/include/tim/vx/ops/localresponsenormalization.h b/include/tim/vx/ops/localresponsenormalization.h index 237151e..d425193 100644 --- a/include/tim/vx/ops/localresponsenormalization.h +++ b/include/tim/vx/ops/localresponsenormalization.h @@ -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) * ``` */