Merge branch 'elem' of https://github.com/xadupre/onnxruntime-extensions

into elem
xadupre · Jun 12, 2024 · 76a8857 · 76a8857
2 parents a4a24a5 + 3bf3cc9
commit 76a8857
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/operators/cuda/add_mul.h b/operators/cuda/add_mul.h
@@ -36,6 +36,7 @@ inline void _FillOutputShape3Op(std::vector<int64_t>& dimsA,
 *  The operator supports broadcast on first dimensions.
 *  A[1, J] + B[I, J] is supported,
 *  A[1, J, 1] + B[I, J, K] is not supported,
+*  In all other cases, all tensors must have the same shape.
 */
 template <typename T, bool addition>
 struct AddOrMulSharedInput {