Apply suggestions from code review

xadupre · Jun 12, 2024 · 86a506c · 86a506c
1 parent 960791b
commit 86a506c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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 <typename T>
 struct MulSigmoid {
@@ -36,6 +38,8 @@ struct MulSigmoid {
 
 /**
 * MulSigmoid(X, Y) = X * Y * Sigmoid(Y)
+
+No shape broadcasting supported.
 */
 template <typename T>
 struct MulMulSigmoid {
@@ -65,4 +69,4 @@ struct MulMulSigmoid {
   }
 };
 
-}  // namespace contrib
+}  // namespace contrib