Skip to main content

primitive_residual_spec

Function primitive_residual_spec 

Source
pub fn primitive_residual_spec(op: &StdTensorOp) -> Option<ResidualSpec>
Expand description

Return the tensor retention declaration for a registered core primitive. Extension declarations remain owned by their context’s rule set.

§Examples

use tenferro_ops::{ad::primitive_residual_spec, std_tensor_op::StdTensorOp};
assert!(primitive_residual_spec(&StdTensorOp::Add).unwrap().is_empty());