forked from mindspore-Ecosystem/mindspore
!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:
commit
09b6b35ad3
|
@ -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}}};
|
||||
|
|
|
@ -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 {
|
||||
|
|
Loading…
Reference in New Issue