Skip to content
Snippets Groups Projects
Commit 48e8ccb6 authored by Tianqi Chen's avatar Tianqi Chen Committed by GitHub
Browse files

[API] Expose AutoInlineInjective (#368)

parent b3bb8126
No related branches found
No related tags found
No related merge requests found
......@@ -18,6 +18,12 @@ TVM_REGISTER_API("schedule.AutoInlineElemWise")
AutoInlineElemWise(args[0]);
});
TVM_REGISTER_API("schedule.AutoInlineInjective")
.set_body([](TVMArgs args, TVMRetValue* ret) {
AutoInlineInjective(args[0]);
});
#define REGISTER_SCHEDULE_PASS1(PassName) \
TVM_REGISTER_API("schedule."#PassName) \
.set_body([](TVMArgs args, TVMRetValue *ret) { \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment