input_shape_attr.getValues<int32_t>().begin(),
                         input_shape_attr.getValues<int32_t>().end());
    } else {
      auto pack = op.getInputSizes().template getDefiningOp<TF::PackOp>();
      if (!pack || pack.getAxis() != 0) return failure();
      auto pack_ty = mlir::dyn_cast<RankedTensorType>(pack.getType());
      if (!pack_ty || pack_ty.getRank() != 1) return failure();