Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dpack_fp16.c77 int batch_oc_offset = b * plane * channel; in PackNHWCToNC8HW8NotAlignedFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dpack_fp32.c41 int batch_oc_offset = b * plane * channel; in PackNHWCToNC4HW4NotAlignedFp32() local
72 int batch_oc_offset = b * plane * channel; in PackNHWCToNC8HW8NotAlignedFp32() local