Home
last modified time | relevance | path

Searched refs:input_y_stride (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c68 int input_y_stride = j * in_w * in_channel + input_stride; in Im2ColPackUnitFp16() local
69 int input_x_stride = input_y_stride + kw_s * in_channel; in Im2ColPackUnitFp16()
76 int input_y_stride = j * dilation_h * in_w * in_channel + input_stride; in Im2ColPackUnitFp16() local
78 int input_x_stride = input_y_stride + n * dilation_w * in_channel; in Im2ColPackUnitFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c57 int input_y_stride = j * in_w * in_channel + input_stride; in Im2ColPackUnitFp32() local
58 int input_x_stride = input_y_stride + kw_s * in_channel; in Im2ColPackUnitFp32()
65 int input_y_stride = j * dilation_h * in_w * in_channel + input_stride; in Im2ColPackUnitFp32() local
67 int input_x_stride = input_y_stride + k * dilation_w * in_channel; in Im2ColPackUnitFp32()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c770 int input_y_stride = j * in_w * in_channel + input_stride; in Im2ColPackUnitInt8Opt() local
771 int input_x_stride = input_y_stride + kw_s * in_channel; in Im2ColPackUnitInt8Opt()
777 int input_y_stride = j * dilation_h * in_w * in_channel + input_stride; in Im2ColPackUnitInt8Opt() local
779 int input_x_stride = input_y_stride + k * dilation_w * in_channel; in Im2ColPackUnitInt8Opt()