Skip to main content

apply_eager

Function apply_eager 

Source
pub fn apply_eager(
    engine: &mut dyn ExecutionEngine,
    op: Arc<dyn ExtensionOp>,
    inputs: &[&EagerTensor],
) -> Result<Vec<EagerTensor>>
Expand description

Apply an extension op to eager tensors on engine, recording it for AD.

§Errors

Returns an input-count or context-mismatch error, a runtime error when no installed extension module serves the op on this engine, or the extension kernel’s own typed error.