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