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:
Родитель
4d051b854b
Коммит
d1d267a09e
|
@ -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 {{
|
||||
|
|
Загрузка…
Ссылка в новой задаче