Searched defs:inter_op_parallel_num (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | thread_pool_reuse_manager.cc | 40 ThreadPool *ThreadPoolReuseManager::GetThreadPool(size_t actor_num, size_t inter_op_parallel_num, s… in GetThreadPool()
64 void ThreadPoolReuseManager::RetrieveThreadPool(size_t actor_num, size_t inter_op_parallel_num, siz… in RetrieveThreadPool()
79 std::string ThreadPoolReuseManager::ComputeHash(size_t actor_num, size_t inter_op_parallel_num, siz… in ComputeHash()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mock/lite_runtime/ |
| D | converters.cc | 28 void ContextUtils::SetContextAttr(int32_t thread_num, int32_t inter_op_parallel_num, in SetContextAttr()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/python/api/ |
| D | context.py | 479 def inter_op_parallel_num(self): member in _CPU
484 def inter_op_parallel_num(self, cpu_inter_op_parallel_num): member in _CPU
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/ |
| D | converters.cc | 27 void ContextUtils::SetContextAttr(int32_t thread_num, int32_t inter_op_parallel_num, bool enable_pa… in SetContextAttr()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | actor_common.cc | 57 …auto inter_op_parallel_num = static_cast<size_t>(context_ptr->get_param<uint32_t>(MS_CTX_INTER_OP_… in ComputeThreadNums() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ |
| D | context.py | 607 def set_inter_op_parallel_num(self, inter_op_parallel_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/native/ |
| D | ms_context.cpp | 603 auto inter_op_parallel_num = c_context_ptr->GetInterOpParallelNum(); in Java_com_mindspore_config_MSContext_getInterOpParallelNum() local
|