Add missing documentation for fused kernels (#744)
This commit is contained in:
Родитель
2f9595af74
Коммит
05df33b302
|
@ -8,6 +8,9 @@
|
|||
|
||||
namespace contrib {
|
||||
|
||||
/**
|
||||
* NegXPlus1(X) = 1 - X
|
||||
*/
|
||||
template <typename T>
|
||||
struct NegXPlus1 {
|
||||
template <typename TDict>
|
||||
|
|
|
@ -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) {
|
||||
|
|
|
@ -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>
|
||||
|
|
Загрузка…
Ссылка в новой задаче