Skip to main content

apply_standard_op

Function apply_standard_op 

Source
pub fn apply_standard_op(
    op: StdTensorOp,
    inputs: &[&EagerTensor],
) -> Result<EagerTensor>
Expand description

Apply one standard tensor op eagerly and record it for AD when needed.

Extension crates use this when an extension-level eager operation expands into ordinary StdTensorOp nodes instead of a custom extension primitive.