Ilya Churaev
0c9abf43a9
* Updated copyright headers
* Revert "Fixed linker warnings in docs snippets on Windows (#15119)"
This reverts commit 372699ec49.
40 lines
1.2 KiB
C++
40 lines
1.2 KiB
C++
// Copyright (C) 2018-2023 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <map>
|
|
|
|
#include "auto_schedule.hpp"
|
|
#include "executable_network.hpp"
|
|
|
|
#ifdef MULTIUNITTEST
|
|
#define MOCKTESTMACRO virtual
|
|
#define MultiDevicePlugin MockMultiDevicePlugin
|
|
#else
|
|
#define MOCKTESTMACRO
|
|
#endif
|
|
|
|
namespace MultiDevicePlugin {
|
|
class AutoExecutableNetwork : public ExecutableNetwork {
|
|
friend IInferPtr AutoSchedule::CreateInferRequest();
|
|
public:
|
|
using Ptr = std::shared_ptr<AutoExecutableNetwork>;
|
|
|
|
explicit AutoExecutableNetwork(AutoScheduleContext::Ptr& context, const AutoSchedule::Ptr& schedule);
|
|
|
|
void SetConfig(const std::map<std::string, IE::Parameter>& config) override;
|
|
IE::Parameter GetConfig(const std::string& name) const override;
|
|
IE::Parameter GetMetric(const std::string& name) const override;
|
|
std::shared_ptr<IE::RemoteContext> GetContext() const override;
|
|
virtual ~AutoExecutableNetwork() = default;
|
|
|
|
private:
|
|
AutoScheduleContext::Ptr _autoSContext;
|
|
AutoSchedule::Ptr _autoSchedule;
|
|
};
|
|
} // namespace MultiDevicePlugin
|