blob: d6a56590d388f6647abdd5517a627ae2efd0d050 [file] [log] [blame]
// Copyright 2018 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef ML_MODEL_METADATA_H_
#define ML_MODEL_METADATA_H_
#include <map>
#include <string>
#include "ml/mojom/model.mojom.h"
namespace ml {
// The information about one supported model.
struct ModelMetadata {
chromeos::machine_learning::mojom::ModelId id;
std::string model_file;
// As accepted by the constructor of ModelImpl.
std::map<std::string, int> required_inputs;
std::map<std::string, int> required_outputs;
};
// Returns a map from model ID to model metdata for each supported model.
std::map<chromeos::machine_learning::mojom::ModelId, ModelMetadata>
GetModelMetadata();
} // namespace ml
#endif // ML_MODEL_METADATA_H_