pub fn build_norm_fwd_graph(
mode: NormMode,
phase: NormPhase,
dtype: DtypeTag,
dims: &[i64],
layout: TensorLayout,
epsilon: f64,
exp_avg_factor: f64,
) -> OperationGraphSpecExpand description
Build the spec-side norm-fwd op graph.