From 86a506cf530799900157264f953b4ac6c524e868 Mon Sep 17 00:00:00 2001 From: Wei-Sheng Chin Date: Tue, 11 Jun 2024 19:57:01 -0700 Subject: [PATCH] Apply suggestions from code review --- operators/cuda/mul_sigmoid.h | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/operators/cuda/mul_sigmoid.h b/operators/cuda/mul_sigmoid.h index 1eb8f2702..a6ddcb74d 100644 --- a/operators/cuda/mul_sigmoid.h +++ b/operators/cuda/mul_sigmoid.h @@ -10,6 +10,8 @@ namespace contrib { /** * MulSigmoid(X) = X * Sigmoid(X) + +No shape broadcasting supported. */ template struct MulSigmoid { @@ -36,6 +38,8 @@ struct MulSigmoid { /** * MulSigmoid(X, Y) = X * Y * Sigmoid(Y) + +No shape broadcasting supported. */ template struct MulMulSigmoid { @@ -65,4 +69,4 @@ struct MulMulSigmoid { } }; -} // namespace contrib \ No newline at end of file +} // namespace contrib