Add the implementation to GetExecGraphInfo API in AUTO plugin (#16979)
This commit is contained in:
@@ -34,4 +34,11 @@ void ExecutableNetwork::SetExeNetworkForContext() {
|
||||
std::string ExecutableNetwork::GetLogTag() const noexcept {
|
||||
return _sContext->_LogTag;
|
||||
}
|
||||
|
||||
std::shared_ptr<ngraph::Function> ExecutableNetwork::GetExecGraphInfo() {
|
||||
// TODO: When there are more then 2 executable networks, such as MULTI:GPU,CPU
|
||||
auto autoSContext = std::dynamic_pointer_cast<AutoScheduleContext>(_sContext);
|
||||
return autoSContext->_hwExecutableNetwork->GetExecGraphInfo();
|
||||
}
|
||||
|
||||
} // namespace MultiDevicePlugin
|
||||
|
||||
@@ -20,6 +20,7 @@ public:
|
||||
using Ptr = std::shared_ptr<ExecutableNetwork>;
|
||||
ExecutableNetwork(const Schedule::Ptr& schedule, const ScheduleContext::Ptr& sContext);
|
||||
IInferPtr CreateInferRequest() override;
|
||||
std::shared_ptr<ngraph::Function> GetExecGraphInfo() override;
|
||||
~ExecutableNetwork() override;
|
||||
|
||||
protected:
|
||||
|
||||
Reference in New Issue
Block a user