const TF::ResourceAliasAnalysis::Info& alias_analysis)
      : alias_analysis_(alias_analysis) {}

  // Returns the recorded device assignment for a resource, if any.
  std::optional<StringRef> DeviceForResource(Value resource) const {
    std::optional<StringRef> result;
    if (alias_analysis_.IsUnknownResource(resource)) return std::nullopt;
    for (int64_t id : alias_analysis_.GetResourceUniqueIds(resource)) {