inference-engine/include/ie_tensor_info.hpp

   1 // Copyright (C) 2018-2020 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 /**
   6  * @brief A header file for the TensorInfo structure
   7  *
   8  * @file ie_tensor_info.hpp
   9  */
  10
  11 #pragma once
  12
  13 #include <map>
  14 #include <memory>
  15 #include <string>
  16
  17 namespace InferenceEngine {
  18
  19 /**
  20  * @struct TensorInfo
  21  * @brief This structure describes tensor information
  22  */
  23 struct TensorInfo {
  24     /**
  25      * @brief A shared pointer to the TensorInfo object
  26      */
  27     using Ptr = std::shared_ptr<TensorInfo>;
  28
  29     /**
  30      * @brief A map of extra info:
  31      * - memory layout BFYX, BXYF (enum)
  32      * - size
  33      * - precision
  34      */
  35     std::map<std::string, std::string> extraInfo;
  36 };
  37
  38 }  // namespace InferenceEngine