define_extension_runtime!() { /* proc-macro */ }Expand description
Generate a standard extension module, preparation engine, and prepared operation.
The execute function must have this signature:
fn<B: BackendBound + 'static>(&OpType, &[&Tensor], &mut ExtensionExecutionContext<'_, B>).
execute_reads is required. It must have this signature:
fn<B: BackendBound + 'static>(&OpType, &[TensorRead<'_>], &mut ExtensionExecutionContext<'_, B>).
session_supported and execute_in_session are optional, but must be
supplied together. The former has signature
fn<B: BackendBound + 'static>(&OpType) -> bool; the latter has signature
fn(&OpType, &mut dyn BackendSession, &mut ExtensionCacheStore, &[TensorRead<'_>]).