if (op.getManualSharding()) {
      // This rewrite does not support manual_sharding. It is expected that the
      // _XlaHostComputeMlirOp registered as an MlirXlaOpKernel will handle this
      // case later once the XlaBuilder graph reaches it.
      return failure();
    }
    return success();
  }
  void rewrite(TF::_XlaHostComputeMlirOp op,
               PatternRewriter& rewriter) const override {