pub fn linearize(
op: &StdTensorOp,
builder: &mut FragmentBuilder<StdTensorOp>,
primal_in: &[GlobalValKey<StdTensorOp>],
primal_out: &[GlobalValKey<StdTensorOp>],
tangent_in: &[Option<LocalValId>],
ctx: &mut ShapeGuardContext,
) -> Vec<Option<LocalValId>>