!15237 mindrt modify include .h

From: @ling_qiao_min
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-05-29 10:21:47 +08:00 committed by Gitee
commit 09b6b35ad3
2 changed files with 2 additions and 0 deletions

View File

@ -65,6 +65,7 @@ struct DeviceContext {
struct Context {
String vendor_name_;
int thread_num_ = 2; /**< thread number config for thread pool */
bool enable_parallel_ = false;
AllocatorPtr allocator = nullptr;
#ifndef NOT_USE_STL
DeviceContextVector device_list_ = {{DT_CPU, {false, MID_CPU}}};

View File

@ -27,6 +27,7 @@ struct MS_API Model {
Uint32Vector input_indices_;
Uint32Vector output_indices_;
int quant_type_;
int device_type_;
};
using NodePtrVector = Vector<Node *>;
struct SubGraph {