inference-engine/src/mkldnn_plugin/mkldnn_async_infer_request.h

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <string>
   8 #include <map>
   9 #include <cpp_interfaces/impl/ie_infer_async_request_thread_safe_default.hpp>
  10 #include "mkldnn_infer_request.h"
  11
  12 namespace MKLDNNPlugin {
  13
  14 class MKLDNNAsyncInferRequest : virtual public InferenceEngine::AsyncInferRequestThreadSafeDefault {
  15 public:
  16     MKLDNNAsyncInferRequest(const InferenceEngine::InferRequestInternal::Ptr &inferRequest,
  17                             const InferenceEngine::ITaskExecutor::Ptr &taskExecutor,
  18                             const InferenceEngine::TaskSynchronizer::Ptr &taskSynchronizer,
  19                             const InferenceEngine::ITaskExecutor::Ptr &callbackExecutor);
  20
  21     ~MKLDNNAsyncInferRequest() override;
  22
  23     void Infer() override;
  24 };
  25
  26 }  // namespace MKLDNNPlugin