inference-engine/include/ie_primitive_info.hpp

   1 // Copyright (C) 2018 Intel Corporation
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4 //
   5
   6 /**
   7  * @brief A header file for the PrimitiveInfo struct
   8  * @file ie_primitive_info.hpp
   9  */
  10
  11 #pragma once
  12
  13 #include "ie_tensor_info.hpp"
  14 #include <string>
  15 #include <vector>
  16 #include <memory>
  17 #include <map>
  18
  19 namespace InferenceEngine {
  20
  21 struct PrimitiveInfo {
  22     using Ptr = std::shared_ptr<PrimitiveInfo>;
  23
  24     std::string sId;          // some internal id, could be used as a name
  25     std::string sType;        // implementation type of this kernel
  26     int iPreAllocatedMemory;  // mainly the allocation of the output tensor
  27
  28     std::vector<TensorInfo::Ptr> inputs;
  29     std::vector<TensorInfo::Ptr> outputs;
  30
  31     std::map<std::string, std::string> extraInfo;  // any other important textual information user might find interesting about this kernel
  32 };
  33
  34 }  // namespace InferenceEngine