Home
last modified time | relevance | path

Searched defs:enable_parallel (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dadaptive_max_pool_3d_grad.cc257 const bool enable_parallel = ctx.Output(0)->GetDataSize() > kParallelDataSize; in AdaptiveMaxPool3dGradCompute() local
Dpad_v3.cc519 const bool enable_parallel = data_num > kParallelNum; in DoCompute() local
540 const bool enable_parallel = data_num > kParallelNum; in DoCompute() local
561 const bool enable_parallel = data_num > kParallelNum; in DoCompute() local
Dpad_v3_grad.cc324 const bool enable_parallel = parallelSliceNum > kParallelNum; in PadV3GradCompute() local
Dadaptive_max_pool_3d.cc239 const bool enable_parallel = (sizeB * sizeD * osizeT * osizeH * osizeW) > kParallelDataNums; in AdaptiveMaxPool3dCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dconverters.cc27 …ntextUtils::SetContextAttr(int32_t thread_num, int32_t inter_op_parallel_num, bool enable_parallel, in SetContextAttr()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/gsm/
Dswap_strategy_builder.cc391 bool enable_parallel = context_->parallel_for_comm_; in HandleFusedTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/native/
Dms_context.cpp29 … jboolean enable_parallel) { in Java_com_mindspore_config_MSContext_createMSContext()