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