Home
last modified time | relevance | path

Searched defs:start_batch (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dmatmul_fp32_avx512.cc108 int start_batch = task_id * batch_stride_; in ParallelRunByBatch() local
229 int start_batch = task_id * batch_stride_; in ParallelRunByGEPDOT() local
274 int start_batch = task_id * batch_stride_; in ParallelRunByRow1Deep1GEPDOT() local
320 int start_batch = task_id * batch_stride_; in ParallelRunByGEPM() local
372 int start_batch = task_id * batch_stride_; in ParallelRunByGEMM() local
448 int start_batch = task_id * batch_stride_; in ParallelRunByBatchColRowGEMM() local
Dmatmul_fp32_sse.cc40 int start_batch = task_id * batch_stride_; in ParallelRunByBatch() local
Dmatmul_fp32_arm32.cc40 int start_batch = task_id * batch_stride_; in ParallelRunByBatch() local
Dmatmul_fp32_avx.cc42 int start_batch = task_id * batch_stride_; in ParallelRunByBatch() local
Dmatmul_fp32_arm64.cc84 int start_batch = task_id * batch_stride_; in ParallelRunByBatch() local
Dmatmul_fp32_base.cc90 int start_batch = task_id * batch_stride_; in ParallelRunByBatch() local
367 int start_batch = task_id * batch_stride_; in ParallelRunIsNotPackByBatch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dmatmul_avx512.c401 int start_batch = task_id * compute->batch_stride_; in MatmulAVX512ParallelRunByBatch() local
444 int start_batch = task_id * matmul->compute_.batch_stride_; in MatmulAVX512ParallelRunByGEPM() local
491 int start_batch = task_id * matmul->compute_.batch_stride_; in MatmulAVX512ParallelRunByGEMM() local
555 int start_batch = task_id * compute->batch_stride_; in MatmulAVX512ParallelRunByGEPDOT() local
603 int start_batch = task_id * matmul->compute_.batch_stride_; in MatmulAVX512ParallelRunByRow1Deep1GEPDOT() local
650 int start_batch = task_id * matmul->compute_.batch_stride_; in MatmulAVX512ParallelRunByBatchColRowGEMM() local
Dmatmul_arm32.c39 int start_batch = task_id * compute->batch_stride_; in MatmulARM32ParallelRunByBatch() local
Dmatmul_sse.c39 int start_batch = task_id * compute->batch_stride_; in MatmulSSEParallelRunByBatch() local
Dmatmul_avx.c39 int start_batch = task_id * compute->batch_stride_; in MatmulAVXParallelRunByBatch() local
Dmatmul_base.c264 int start_batch = task_id * compute->batch_stride_; in MatmulBaseParallelRunByBatch() local
291 int start_batch = task_id * matmul->compute_.batch_stride_; in MatmulBaseParallelRunIsNotPackByBatch() local
Dmatmul_arm64.c122 int start_batch = task_id * compute->batch_stride_; in MatmulARM64ParallelRunByBatch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dpack_fp32_opt.c151 int64_t start_batch = start / unit_num_per_batch; in RowMajor2Col12MajorOpt() local
225 int64_t start_batch = start / unit_num_per_batch; in RowMajor2Row12MajorOpt() local
Dconv_im2col_avx512_fp32.c70 int start_batch = block_batch_per_thread * task_id; in ConvIm2ColAVX512Fp32CutByBatch() local
Dconv_winograd_fp32.c179 int start_batch = block_batch_per_thread * task_id; in ConvWinogardFp32CutByBatch() local
Dconv_common_fp32.c199 int start_batch = block_batch_per_thread * task_id; in ConvFp32CutByBatch() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Darithmetic_fp32_coder.cc369 int start_batch = batch_per_thread * task_id; in BatchScalarCalc() local
404 int start_batch = batch_per_thread * task_id; in BiasCalc() local
/third_party/grpc/src/core/lib/surface/
Dfilter_stack_call.cc315 grpc_closure start_batch; member