Add missing documentation for fused kernels (#744)

This commit is contained in:
Xavier Dupré 2024-06-18 10:54:15 +02:00 коммит произвёл GitHub
Родитель 2f9595af74
Коммит 05df33b302
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
3 изменённых файлов: 15 добавлений и 0 удалений

Просмотреть файл

@ -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>