Struct CustomData 

pub struct CustomData(pub Arc<dyn CustomOp>);

Wrapper around Arc<dyn CustomOp> that implements Debug for Op.

Tuple Fields

0: Arc<dyn CustomOp>

Implementations

impl CustomData

pub fn new<T: CustomOp>(op: T) -> Self

Wraps a CustomOp implementation.

pub fn name(&self) -> &str

The wrapped op’s name.

pub fn infer_output_shape(&self, input_shapes: &[&Shape]) -> Shape

The wrapped op’s inferred output shape.

pub fn downcast_ref<T: Any>(&self) -> Option<&T>

Downcast to a concrete op type via CustomOp::as_any.

Trait Implementations

impl Clone for CustomData

fn clone(&self) -> CustomData

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CustomData

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.