Update Custom Op Template Gen based on recent changes (#371)

* Added initial CustomOp template generator

* Added arg parsing, error check, and modularized methods

* Added test and models for the same plus refactored code in template generator

* Fixed model path issue

* Final changes

* Removed ONNX models and added model generation in test scripts, changed custom ops to random math ops

* Update based on new custom op changes

---------

Co-authored-by: Sayan Shaw <sayanshaw@microsoft.com>
This commit is contained in:
Sayan Shaw 2023-02-27 16:59:31 -08:00 коммит произвёл GitHub
Родитель 4d051b854b
Коммит d1d267a09e
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 0 добавлений и 4 удалений

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

@ -42,10 +42,6 @@ def create_hpp(customop_template_filepath, op, op_name, input_type_count, output
#include "string_utils.h"
#include "ustring.h"
void* {custom_op}::CreateKernel(const OrtApi& api, const OrtKernelInfo* info) const {{
return CreateKernelImpl(api, info);
}};
const char* {custom_op}::GetName() const {{ return "{op_name}"; }};
size_t {custom_op}::GetInputTypeCount() const {{