pub fn to_axis_ops_with_tf_rules( input_orig: &[TDim], output_spec: &[TDim], ) -> Result<SmallVec<[AxisOp; 4]>, Error>