Home
last modified time | relevance | path

Searched refs:src_plane_offset (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dwinograd_transform.c44 int src_plane_offset = in_channel * (src_y_s * input_w + src_x_s); in WinogradInputTransform() local
52 int src_ic4_offset = src_plane_offset + ic * C4NUM; in WinogradInputTransform()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_transform_fp16.c49 int src_plane_offset = in_channel * (src_y_s * input_w + src_x_s); in WinogradInputTransformFp16() local
57 int src_ic8_offset = src_plane_offset + ic * C8NUM; in WinogradInputTransformFp16()
Dpack_fp16.c334 int src_plane_offset = src_c_offset + i; in PackNCHWToNHWC4Fp16() local
336 ((float16_t *)dst)[dst_plane_offset] = ((float16_t *)src)[src_plane_offset]; in PackNCHWToNHWC4Fp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c825 int src_plane_offset = src_batch_offset + k * in_channel; in PackInputToC8Int8() local
828 int src_c_offset = src_plane_offset + i; in PackInputToC8Int8()
841 int src_c_offset = src_plane_offset + ic8 + l; in PackInputToC8Int8()
Dconv3x3_int8.c805 int src_plane_offset = C8NUM * (origin_y * input_width + origin_x); in Conv3x3Int8InputTransform() local
811 int src_c8_offset = src_plane_offset + ic * C8NUM * input_height * input_width; in Conv3x3Int8InputTransform()