Struct CustomOpRegistry 

pub struct CustomOpRegistry { /* private fields */ }

Registry for custom operations.

Implementations

impl CustomOpRegistry

pub fn new() -> Self

Create a new registry.

pub fn register( &mut self, name: impl Into<String>, metadata: CustomOpMetadata, handler: CustomOpHandler, ) -> Result<()>

Register a custom operation.

Examples

use tensorlogic_compiler::compile::CustomOpRegistry;
use tensorlogic_compiler::compile::CustomOpMetadata;
use std::sync::Arc;

let mut registry = CustomOpRegistry::new();

let metadata = CustomOpMetadata {
    name: "custom_and".to_string(),
    description: "Custom AND with threshold".to_string(),
    expected_arity: Some(2),
    is_differentiable: true,
};

registry.register(
    "custom_and",
    metadata,
    Arc::new(|expr, ctx, graph, data| {
        // Custom compilation logic here
        Ok(0)
    }),
).unwrap();

pub fn unregister(&mut self, name: &str) -> Result<()>

Unregister a custom operation.

pub fn has_operation(&self, name: &str) -> bool

Check if an operation is registered.

pub fn get_metadata(&self, name: &str) -> Option<CustomOpMetadata>

Get metadata for an operation.

pub fn list_operations(&self) -> Vec<String>

List all registered operations.

pub fn invoke( &self, name: &str, expr: &TLExpr, ctx: &mut CompilerContext, graph: &mut EinsumGraph, data: &CustomOpData, ) -> Result<usize>

Invoke a custom operation.

Trait Implementations

impl Default for CustomOpRegistry

fn default() -> Self

Returns the “default value” for a type.