Previously we support a limited case of function type deduction and in many places we have to supply the type twice during set_body_typed (one in the template parameter, another in the lambda signature). This PR improves the deduce function by enablng automatic function signature deduction. ``` TVM_REGISTER_GLOBAL("sub") .set_body_typed([](int x, int y) -> int { return x - y; }); ``` Unfortunately, because of template conflict, we can not support the original case where both type signature and lambda are supplied through set_body_typed. This PR refactors the existing regsitration to the new style.
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
quantize | Loading commit data... | |
alter_op_layout.cc | Loading commit data... | |
canonicalize_cast.cc | Loading commit data... | |
canonicalize_ops.cc | Loading commit data... | |
combine_parallel_conv2d.cc | Loading commit data... | |
combine_parallel_dense.cc | Loading commit data... | |
combine_parallel_op.cc | Loading commit data... | |
combine_parallel_op.h | Loading commit data... | |
combine_parallel_op_batch.cc | Loading commit data... | |
combine_parallel_op_batch.h | Loading commit data... | |
convert_layout.cc | Loading commit data... | |
de_duplicate.cc | Loading commit data... | |
dead_code.cc | Loading commit data... | |
dependency_graph.cc | Loading commit data... | |
dependency_graph.h | Loading commit data... | |
device_annotation.cc | Loading commit data... | |
eliminate_common_subexpr.cc | Loading commit data... | |
eta_expand.cc | Loading commit data... | |
expr_subst.cc | Loading commit data... | |
expr_subst.h | Loading commit data... | |
feature.cc | Loading commit data... | |
fold_constant.cc | Loading commit data... | |
fold_scale_axis.cc | Loading commit data... | |
forward_rewrite.cc | Loading commit data... | |
fuse_ops.cc | Loading commit data... | |
gradient.cc | Loading commit data... | |
infer_layout_util.h | Loading commit data... | |
kind_check.cc | Loading commit data... | |
legalize.cc | Loading commit data... | |
let_list.h | Loading commit data... | |
mac_count.cc | Loading commit data... | |
match_exhaustion.cc | Loading commit data... | |
partial_eval.cc | Loading commit data... | |
pass_manager.cc | Loading commit data... | |
pass_util.h | Loading commit data... | |
pattern_util.h | Loading commit data... | |
print_ir.cc | Loading commit data... | |
simplify_inference.cc | Loading commit data... | |
to_a_normal_form.cc | Loading commit data... | |
to_cps.cc | Loading commit data... | |
to_graph_normal_form.cc | Loading commit data... | |
transform_layout.h | Loading commit data... | |
type_infer.cc | Loading commit data... | |
type_solver.cc | Loading commit data... | |
type_solver.h | Loading commit data... | |
util.cc | Loading commit data... | |
well_formed.cc | Loading commit data... |