Skip to main content

apply_expanded_graph

Function apply_expanded_graph 

Source
pub fn apply_expanded_graph(
    inputs: &[&TracedTensor],
    output_metas: Vec<(DType, Vec<SymDim>)>,
    build: impl FnOnce(&mut GraphBuilder<StdTensorOp>, &[ValueRef<StdTensorOp>]) -> Result<Vec<usize>>,
) -> Result<Vec<TracedTensor>>
Expand description

Apply an extension-provided lowering as ordinary traced graph operations.

This is for extension crates whose operation can be expanded at graph-build time. It preserves the same parent graph and metadata merging behavior as apply, but does not insert a StdTensorOp::Extension carrier.

ยงErrors

Returns Error::Validation with InvalidArgument when lowering produces an invalid output count or unknown output metadata, Error::Internal for an invalid graph reference, and Error::RuntimeStateSource when metadata registration cannot retain the lowered graph state.