#ifndef COMPONENTS_ML_MOJOM_WEB_PLATFORM_MODEL_MOJOM_H_
#define COMPONENTS_ML_MOJOM_WEB_PLATFORM_MODEL_MOJOM_H_
#include <stdint.h>
#include <limits>
#include <optional>
#include <type_traits>
#include <utility>
#include "base/types/cxx23_to_underlying.h"
#include "mojo/public/cpp/bindings/clone_traits.h"
#include "mojo/public/cpp/bindings/equals_traits.h"
#include "mojo/public/cpp/bindings/lib/serialization.h"
#include "mojo/public/cpp/bindings/struct_ptr.h"
#include "mojo/public/cpp/bindings/struct_traits.h"
#include "mojo/public/cpp/bindings/union_traits.h"
#include "third_party/perfetto/include/perfetto/tracing/traced_value_forward.h"
#include "components/ml/mojom/web_platform_model.mojom-features.h"
#include "components/ml/mojom/web_platform_model.mojom-shared.h"
#include "components/ml/mojom/web_platform_model.mojom-forward.h"
#include "mojo/public/mojom/base/big_buffer.mojom.h"
#include <string>
#include <vector>
#include "mojo/public/cpp/bindings/lib/control_message_handler.h"
#include "mojo/public/cpp/bindings/lib/message_size_estimator.h"
#include "mojo/public/cpp/bindings/raw_ptr_impl_ref_traits.h"
namespace ml::model_loader::mojom {
class ModelLoaderProxy;
template <typename ImplRefTraits>
class ModelLoaderStub;
class ModelLoaderRequestValidator;
class ModelLoaderResponseValidator;
class ModelLoader
: public ModelLoaderInterfaceBase { … };
class ModelProxy;
template <typename ImplRefTraits>
class ModelStub;
class ModelRequestValidator;
class ModelResponseValidator;
class Model
: public ModelInterfaceBase { … };
class ModelLoaderProxy
: public ModelLoader { … };
class ModelProxy
: public Model { … };
class ModelLoaderStubDispatch { … };
template <typename ImplRefTraits =
mojo::RawPtrImplRefTraits<ModelLoader>>
class ModelLoaderStub
: public mojo::MessageReceiverWithResponderStatus {
public:
using ImplPointerType = typename ImplRefTraits::PointerType;
ModelLoaderStub() = default;
~ModelLoaderStub() override = default;
void set_sink(ImplPointerType sink) { … }
ImplPointerType& sink() { … }
bool Accept(mojo::Message* message) override { … }
bool AcceptWithResponder(
mojo::Message* message,
std::unique_ptr<mojo::MessageReceiverWithStatus> responder) override { … }
private:
ImplPointerType sink_;
};
class ModelStubDispatch { … };
template <typename ImplRefTraits =
mojo::RawPtrImplRefTraits<Model>>
class ModelStub
: public mojo::MessageReceiverWithResponderStatus {
public:
using ImplPointerType = typename ImplRefTraits::PointerType;
ModelStub() = default;
~ModelStub() override = default;
void set_sink(ImplPointerType sink) { … }
ImplPointerType& sink() { … }
bool Accept(mojo::Message* message) override { … }
bool AcceptWithResponder(
mojo::Message* message,
std::unique_ptr<mojo::MessageReceiverWithStatus> responder) override { … }
private:
ImplPointerType sink_;
};
class ModelLoaderRequestValidator : public mojo::MessageReceiver { … };
class ModelRequestValidator : public mojo::MessageReceiver { … };
class ModelLoaderResponseValidator : public mojo::MessageReceiver { … };
class ModelResponseValidator : public mojo::MessageReceiver { … };
class CreateModelLoaderOptions { … };
template <typename T, CreateModelLoaderOptions::EnableIfSame<T>* = nullptr>
bool operator<(const T& lhs, const T& rhs);
template <typename T, CreateModelLoaderOptions::EnableIfSame<T>* = nullptr>
bool operator<=(const T& lhs, const T& rhs) { … }
template <typename T, CreateModelLoaderOptions::EnableIfSame<T>* = nullptr>
bool operator>(const T& lhs, const T& rhs) { … }
template <typename T, CreateModelLoaderOptions::EnableIfSame<T>* = nullptr>
bool operator>=(const T& lhs, const T& rhs) { … }
class TensorInfo { … };
template <typename T, TensorInfo::EnableIfSame<T>* = nullptr>
bool operator<(const T& lhs, const T& rhs);
template <typename T, TensorInfo::EnableIfSame<T>* = nullptr>
bool operator<=(const T& lhs, const T& rhs) { … }
template <typename T, TensorInfo::EnableIfSame<T>* = nullptr>
bool operator>(const T& lhs, const T& rhs) { … }
template <typename T, TensorInfo::EnableIfSame<T>* = nullptr>
bool operator>=(const T& lhs, const T& rhs) { … }
class ModelInfo { … };
template <typename T, ModelInfo::EnableIfSame<T>* = nullptr>
bool operator<(const T& lhs, const T& rhs);
template <typename T, ModelInfo::EnableIfSame<T>* = nullptr>
bool operator<=(const T& lhs, const T& rhs) { … }
template <typename T, ModelInfo::EnableIfSame<T>* = nullptr>
bool operator>(const T& lhs, const T& rhs) { … }
template <typename T, ModelInfo::EnableIfSame<T>* = nullptr>
bool operator>=(const T& lhs, const T& rhs) { … }
template <typename StructPtrType>
CreateModelLoaderOptionsPtr CreateModelLoaderOptions::Clone() const { … }
template <typename T, CreateModelLoaderOptions::EnableIfSame<T>*>
bool CreateModelLoaderOptions::Equals(const T& other_struct) const { … }
template <typename T, CreateModelLoaderOptions::EnableIfSame<T>*>
bool operator<(const T& lhs, const T& rhs) { … }
template <typename StructPtrType>
TensorInfoPtr TensorInfo::Clone() const { … }
template <typename T, TensorInfo::EnableIfSame<T>*>
bool TensorInfo::Equals(const T& other_struct) const { … }
template <typename T, TensorInfo::EnableIfSame<T>*>
bool operator<(const T& lhs, const T& rhs) { … }
template <typename StructPtrType>
ModelInfoPtr ModelInfo::Clone() const { … }
template <typename T, ModelInfo::EnableIfSame<T>*>
bool ModelInfo::Equals(const T& other_struct) const { … }
template <typename T, ModelInfo::EnableIfSame<T>*>
bool operator<(const T& lhs, const T& rhs) { … }
}
namespace mojo {
template <>
struct StructTraits<::ml::model_loader::mojom::CreateModelLoaderOptions::DataView,
::ml::model_loader::mojom::CreateModelLoaderOptionsPtr> { … };
template <>
struct StructTraits<::ml::model_loader::mojom::TensorInfo::DataView,
::ml::model_loader::mojom::TensorInfoPtr> { … };
template <>
struct StructTraits<::ml::model_loader::mojom::ModelInfo::DataView,
::ml::model_loader::mojom::ModelInfoPtr> { … };
}
#endif