return failure();
    }

    Operation* owner_op;
    for (const auto& [attr, val] : attr_to_op_map) {
      if (attr.getName() == attribute.getName()) owner_op = val;
    }
    if (stablehlo::IsStablehloOp(owner_op)) {
      owner_op->setAttr(StringRef(attribute.getName()), attribute.getValue());
    } else {
      owner_op = attr_to_op_map[attribute];

      std::string new_attr_map_str{};

                try {
                    ownerGroup.resolve(server, getContext());
                }
                catch ( IOException e ) {
                    log.warn("Failed to resolve SID " + ownerGroup.toString(), e);
                }
            }
            else {
                ownerGroup.initContext(server, getContext());
            }
            return ownerGroup;
        }
    }