Home
last modified time | relevance | path

Searched defs:parallel_num (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/
Dcontext.cc106 void Context::SetInterOpParallelNum(int32_t parallel_num) { in SetInterOpParallelNum()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dupsample_trilinear3d_grad.cc268 …UpsampleTrilinear3dGradCpuKernel::FetchBlockSize(CpuKernelContext &ctx, const int64_t parallel_num, in FetchBlockSize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dcontext.cc97 void Context::SetInterOpParallelNum(int32_t parallel_num) { in SetInterOpParallelNum()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dresize_bicubic_grad_cpu_kernel.cc304 …const size_t parallel_num = static_cast<size_t>(RGS.batch_size * RGS.channels * RGS.resized_height… in ResizeBicubicGrad() local
Dresize_bicubic_cpu_kernel.cc279 const size_t parallel_num = static_cast<size_t>(out_ch * RS.batch_size); in interpolate_with_caching() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Dembedding_cache_prefetch_actor.cc141 size_t parallel_num = unique_ids->multi_batch_data_.size(); in DeduplicateId() local