assert(port[0] == 0);

  auto params = op.getParams();
  auto params_ty = mlir::dyn_cast<RankedTensorType>(params.getType());
  if (!params_ty || !params_ty.hasStaticShape() || params_ty.getRank() != 1 ||
      op.getBatchDims() != 0) {
    return {};
  }

  DenseIntElementsAttr axis;
  if (!matchPattern(op.getAxis(), m_Constant(&axis)) ||
      axis.getNumElements() != 1 ||
      !axis.getSplatValue<llvm::APInt>().isZero()) {