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