Searched refs:col_algin (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | matmul_fp32.h | 86 size_t row_block, size_t col_block, size_t col_algin, size_t deep);
90 size_t row_block, size_t col_block, size_t col_algin, size_t deep);
92 size_t row_block, size_t col_block, size_t col_algin, size_t deep);
94 size_t row_block, size_t col_block, size_t col_algin, size_t deep);
96 size_t row_block, size_t col_block, size_t col_algin, size_t deep);
101 size_t row_block, size_t col_block, size_t col_algin, size_t deep);
|
| D | matmul_fp32.c | 1096 size_t row_block, size_t col_block, size_t col_algin, size_t deep) { in MatVecMul1x32Kernel() argument
1211 size_t row_block, size_t col_block, size_t col_algin, size_t deep) { in MatVecMul1x24Kernel() argument
1316 size_t row_block, size_t col_block, size_t col_algin, size_t deep) { in MatVecMul1x16Kernel() argument
1406 size_t row_block, size_t col_block, size_t col_algin, size_t deep) { in MatVecMul1x8Kernel() argument
1476 size_t row_block, size_t col_block, size_t col_algin, size_t deep) { in MatVecMulRowxColKernel() argument
1517 _mm256_storeu_ps(dst + i * col_algin + j * C8NUM, dst_data[i * col_block + j]); in MatVecMulRowxColKernel()
|