ie_tensor_info.hpp

Go to the documentation of this file.

// Copyright (C) 2018-2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

/**
 * @brief A header file for the TensorInfo structure
 *
 * @file ie_tensor_info.hpp
 */

#pragma once

#include <map>
#include <memory>
#include <string>

namespace InferenceEngine {

/**
 * @struct TensorInfo
 * @brief This structure describes tensor information
 */
struct TensorInfo {
    /**
     * @brief A shared pointer to the TensorInfo object
     */
    using Ptr = std::shared_ptr<TensorInfo>;

    /**
     * @brief A map of extra info:
     * - memory layout BFYX, BXYF (enum)
     * - size
     * - precision
     */
    std::map<std::string, std::string> extraInfo;
};

}  // namespace InferenceEngine