HDK
|
#include <cann_provider_options.h>
Public Attributes | |
int | device_id |
size_t | npu_mem_limit |
onnxruntime::ArenaExtendStrategy | arena_extend_strategy |
int | enable_cann_graph |
int | dump_graphs |
std::string | precision_mode |
std::string | op_select_impl_mode |
std::string | optypelist_for_implmode |
OrtArenaCfg * | default_memory_arena_cfg |
Definition at line 12 of file cann_provider_options.h.
onnxruntime::ArenaExtendStrategy OrtCANNProviderOptions::arena_extend_strategy |
Definition at line 15 of file cann_provider_options.h.
OrtArenaCfg* OrtCANNProviderOptions::default_memory_arena_cfg |
Definition at line 24 of file cann_provider_options.h.
int OrtCANNProviderOptions::device_id |
Definition at line 13 of file cann_provider_options.h.
int OrtCANNProviderOptions::dump_graphs |
Definition at line 18 of file cann_provider_options.h.
int OrtCANNProviderOptions::enable_cann_graph |
Definition at line 16 of file cann_provider_options.h.
size_t OrtCANNProviderOptions::npu_mem_limit |
Definition at line 14 of file cann_provider_options.h.
std::string OrtCANNProviderOptions::op_select_impl_mode |
Definition at line 20 of file cann_provider_options.h.
std::string OrtCANNProviderOptions::optypelist_for_implmode |
Definition at line 22 of file cann_provider_options.h.
std::string OrtCANNProviderOptions::precision_mode |
Definition at line 19 of file cann_provider_options.h.