Home
last modified time | relevance | path

Searched defs:matrix_deep (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dmatmul_fp32_avx512.cc271 int matrix_deep = params_->deep_; in ParallelRunByRow1Deep1GEPDOT() local
317 int matrix_deep = params_->deep_; in ParallelRunByGEPM() local
369 int matrix_deep = params_->deep_; in ParallelRunByGEMM() local
445 int matrix_deep = params_->deep_; in ParallelRunByBatchColRowGEMM() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dmatmul_avx512.c441 int matrix_deep = matmul->compute_.deep_; in MatmulAVX512ParallelRunByGEPM() local
488 int matrix_deep = matmul->compute_.deep_; in MatmulAVX512ParallelRunByGEMM() local
600 int matrix_deep = matmul->compute_.deep_; in MatmulAVX512ParallelRunByRow1Deep1GEPDOT() local
647 int matrix_deep = matmul->compute_.deep_; in MatmulAVX512ParallelRunByBatchColRowGEMM() local