3 Commits

Author	SHA1	Message	Date
Laurent Mazare	5cc843550d	Add binary and ternary custom ops. (#217)	2023-07-21 17:29:50 +01:00
Laurent Mazare	4a100875bf	Use a macro to handle the dtype pattern matching. (#215)	2023-07-21 16:03:51 +01:00
Laurent Mazare	a6bcdfb269	Custom ops with a single argument (#214) ... * Add the CustomOp1 trait. * Add an example of custom op. * Polish the custom op example. * Add some backward pass test for custom ops.	2023-07-21 15:18:05 +01:00