Commit Graph

10 Commits

Author SHA1 Message Date
looop5 ae5532fb9e pass lambda function to out_shape and out_dtype 2021-12-06 10:21:16 +08:00
Yang Jiao 5acce31c72 apply autodiff in custom 2021-12-03 11:16:19 +08:00
looop5 9c4fe8b6bf update akg, refine document of Custom op 2021-11-16 11:06:40 +08:00
looop5 58e27d87bc add Custom, custom_op_info_register, CustomRegOp to __init__ 2021-11-11 20:49:16 +08:00
Yang Jiao 2b112ef656 add custom pynative test cases 2021-11-08 17:37:39 +08:00
looop5 b89d744e80 Custom op supports no reg info 2021-11-05 17:21:39 +08:00
Yang Jiao 9bcf28d15d adapt custom op to pyfunc kernel 2021-10-31 22:08:10 +08:00
looop5 0b34bf33bc add custom op pynative testcases 2021-10-25 20:53:42 +08:00
Yang Jiao 40b648b873 add aot 2021-10-22 11:37:39 +08:00
looop5 46789f260a Custom operator supports tbe dsl 2021-10-15 16:28:33 +08:00