Add missing documentation for fused kernels

xadupre · Jun 11, 2024 · d3b6f5d · d3b6f5d
1 parent f505546
commit d3b6f5d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/operators/cuda/negxplus1.h b/operators/cuda/negxplus1.h
@@ -8,6 +8,9 @@
 
 namespace contrib {
 
+/**
+* NegXPlus1(X) = 1 - X
+*/
 template <typename T>
 struct NegXPlus1 {
   template <typename TDict>

diff --git a/operators/cuda/scatter_nd_of_shape.h b/operators/cuda/scatter_nd_of_shape.h
@@ -8,6 +8,9 @@
 
 namespace contrib {
 
+/**
+* ScatterNDOfShape(shape, indices, updates) = ScatterND(ConstantOfShape(shape, value=0), indices, updates)
+*/
 template <typename T>
 struct ScatterNDOfShape {
   OrtStatusPtr OnModelAttach(const OrtApi& api, const OrtKernelInfo& info) {
@@ -71,6 +74,12 @@ struct ScatterNDOfShape {
 };
 
 
+/**
+* MaskedScatterNDOfShape(shape, indices, updates) = ScatterND(ConstantOfShape(shape, value=0),
+*                                                             indices[indices != maskedValue],
+*                                                             updates[indices != maskedValue])
+*
+*/
 template <typename T>
 struct MaskedScatterNDOfShape {
   OrtStatusPtr OnModelAttach(const OrtApi& api, const OrtKernelInfo& info) {

diff --git a/operators/cuda/transpose_cast.h b/operators/cuda/transpose_cast.h
@@ -8,6 +8,9 @@
 
 namespace contrib {
 
+/**
+*  Transpose2DCast(X, to=to) = Cast(Transpose(X, perm=[1, 0]), to=to)
+*/
 template <typename TIN, typename TOUT>
 struct Transpose2DCast {
   template <typename TDict>