Skip to content

Commit 2521dab

Browse files
sayanshaw24Sayan Shaw
authored andcommitted
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>
1 parent 8ccbb8c commit 2521dab

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

tools/gen_customop_template.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,6 @@ def create_hpp(customop_template_filepath, op, op_name, input_type_count, output
4242
#include "string_utils.h"
4343
#include "ustring.h"
4444
45-
void* {custom_op}::CreateKernel(const OrtApi& api, const OrtKernelInfo* info) const {{
46-
return CreateKernelImpl(api, info);
47-
}};
48-
4945
const char* {custom_op}::GetName() const {{ return "{op_name}"; }};
5046
5147
size_t {custom_op}::GetInputTypeCount() const {{

0 commit comments

Comments
 (0)