Method PrimitiveSerializer

mlx/export.cpp:41–49

  using StateExtractor = std::function<std::vector<StateT>(const Primitive&)>;

  PrimitiveSerializer(
      Serializer serialize,
      Deserializer deserialize,
      StateExtractor extract_state,
      std::vector<std::string> keys = {})
      : serialize(std::move(serialize)),
        deserialize(std::move(deserialize)),
        extract_state(std::move(extract_state)),
        keys(std::move(keys)) {};
  Serializer serialize;
  Deserializer deserialize;
  StateExtractor extract_state;