Skip to main content

ad_rules

tenferro_linalg

Function ad_rules 

Source 
pub fn ad_rules() -> Result<ExtensionRuleSet, ExtensionRegistryError>
Expand description

Return the explicit linalg extension AD rule set.

ยงExamples

let rules = tenferro_linalg::ad_rules().unwrap();
assert!(rules.is_linearize_registered(tenferro_linalg::LINALG_EXTENSION_FAMILY_ID));
assert!(rules.is_linear_transpose_registered(tenferro_linalg::LINALG_EXTENSION_FAMILY_ID));