Searched defs:oc_index (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/ |
| D | matmul_fp32_avx.h | 128 static inline int64_t MatVecMulNoPackCoreAVX(int64_t oc_index, const float *a, const float *b, floa… in MatVecMulNoPackCoreAVX()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/ |
| D | matmul_fp32_sse.h | 128 static inline int64_t MatVecMulNoPackCoreSSE(int64_t oc_index, const float *a, const float *b, floa… in MatVecMulNoPackCoreSSE()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/ |
| D | matmul_fp32_avx512.h | 128 static inline int64_t MatVecMulNoPackCoreAVX512(int64_t oc_index, const float *a, const float *b, f… in MatVecMulNoPackCoreAVX512()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | matmul_fp32_neon.h | 127 static inline int64_t MatVecMulNoPackCoreNEON(int64_t oc_index, const float *a, const float *b, flo… in MatVecMulNoPackCoreNEON()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | matmul_fp32.c | 431 int64_t oc_index = 0; in MatVecMulNoPackFp32() local
450 int64_t oc_index = 0; in MatVecMulNoPackFp32() local
|
| D | pack_fp32.c | 1264 int oc_index = oc * kernel_w * kernel_h * input_channel + ic_index; in SWPackNHWCToNXHWCXFp32() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | conv3x3_int8.c | 624 int oc_index = oc_start + i; in Conv3x3Int8OutputUnit() local
|