Skip to content

Commit

Permalink
Update Custom Op Template Gen based on recent changes (#371)
Browse files Browse the repository at this point in the history
* 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 <[email protected]>
  • Loading branch information
2 people authored and wenbingl committed Mar 1, 2023
1 parent 8ccbb8c commit 2521dab
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions tools/gen_customop_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 {{
Expand Down

0 comments on commit 2521dab

Please sign in to comment.