// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2019-2022 Second State INC

#include "tensorflow_env.h"
#include "tensorflow_module.h"

namespace WasmEdge {
namespace Host {

namespace {

Runtime::Instance::ModuleInstance *
create(const Plugin::PluginModule::ModuleDescriptor *) noexcept {
  return new WasmEdgeTensorflowModule;
}

Plugin::Plugin::PluginDescriptor Descriptor{
    .Name = "wasmedge_tensorflow",
    .Description = "Tensorflow plug-in for WasmEdge.",
    .APIVersion = Plugin::Plugin::CurrentAPIVersion,
    .Version = {0, 13, 0, 0},
    .ModuleCount = 1,
    .ModuleDescriptions =
        (Plugin::PluginModule::ModuleDescriptor[]){
            {
                .Name = "wasmedge_tensorflow",
                .Description =
                    "This module contains WasmEdge-Tensorflow host functions.",
                .Create = create,
            },
        },
    .AddOptions = nullptr,
};

} // namespace

Plugin::PluginRegister WasmEdgeTensorflow::TFEnv::Register(&Descriptor);

} // namespace Host
} // namespace WasmEdge
