pub fn compute_shape_with_tf_rules(
    input: &[TDim],
    shape_spec: &[TDim]
) -> Result<SmallVec<[TDim; 4]>, Error>