Home
last modified time | relevance | path

Searched defs:k_index (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dmatmul_fp32.c386 int k_index = 0; in GemmIsNotPackOptimize() local
412 int k_index = 0; in GemmIsNotPackOptimize() local
435 for (int64_t k_index = 0; k_index < k; ++k_index) { in MatVecMulNoPackFp32() local
454 for (int64_t k_index = 0; k_index < depth; ++k_index) { in MatVecMulNoPackFp32() local
Dmatmul_avx512_fp32.c211 int k_index = 0; in GemmIsNotPackOptimizeAVX512() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc3802 int64_t k_index = axis_attr; in ExportPrimSort() local