diff --git a/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl b/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl index 4b07a1e89e0..791ba888eb1 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl +++ b/mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl @@ -58,10 +58,10 @@ __kernel void transpose_NHWC4_BUF(__read_only image2d_t src_data, global FLT4 *d bool over_size = W * C.y > 65535; FLT4 x0, x1, x2, x3; if (over_size) { - x0 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X)); - x1 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X + 1)); - x2 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X + 2)); - x3 = READ_IMAGE(src_data, smp_zero, (int2)(C, 4 * X + 3)); + x0 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X)); + x1 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X + 1)); + x2 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X + 2)); + x3 = READ_IMAGE(src_data, smp_zero, (int2)(Y, 4 * X + 3)); } else { x0 = READ_IMAGE(src_data, smp_zero, (int2)((4 * X) % W * C.y + Y, (4 * X) / W)); x1 = READ_IMAGE(src_data, smp_zero, (int2)((4 * X + 1) % W * C.y + Y, (4 * X + 1) / W)); diff --git a/mindspore/lite/test/models_tflite_gpu.cfg b/mindspore/lite/test/models_tflite_gpu.cfg index d432b29fef9..ac008b9b182 100644 --- a/mindspore/lite/test/models_tflite_gpu.cfg +++ b/mindspore/lite/test/models_tflite_gpu.cfg @@ -1,3 +1,5 @@ mobilenet_v1_1.0_224.tflite mobilenet_v2_1.0_224.tflite -resnet.tflite \ No newline at end of file +resnet.tflite +hiai_cn_recognize_modify_padv2.tflite +hiai_cv_focusShootOCRModel_08.tflite