Skip to main content

cast_params_for_forward

yscv_model

Function cast_params_for_forward 

Source 
pub fn cast_params_for_forward(
    graph: &Graph,
    param_nodes: &[NodeId],
    target_dtype: DType,
) -> Result<Vec<Tensor>, ModelError>
Expand description

Convert model parameters from master precision to forward precision.