Skip to main content

build_norm_fwd_graph

Function build_norm_fwd_graph 

Source
pub fn build_norm_fwd_graph(
    mode: NormMode,
    phase: NormPhase,
    dtype: DtypeTag,
    dims: &[i64],
    layout: TensorLayout,
    epsilon: f64,
    exp_avg_factor: f64,
) -> OperationGraphSpec
Expand description

Build the spec-side norm-fwd op graph.