Method load

mlx/export.cpp:494–503

  }

  std::shared_ptr<Primitive> load(Reader& is) {
    auto stream = resolve_stream(deserialize<Stream>(is));
    auto name = deserialize<std::string>(is);
    if (auto it = factory.find(name); it != factory.end()) {
      return it->second.deserialize(is, stream);
    } else {
      throw std::invalid_argument(
          "[import_function] Unable to deserialize primitive " + name);
    }
  };

  std::pair<std::string, std::vector<StateT>> extract_state(
      const std::shared_ptr<Primitive>& p) {