Browse Source

!5746 [MS][LITE][GPU]GPU CI: add OCR02 and OCR08

Merge pull request !5746 from chenzupeng/master-lite
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
349356d8d7
2 changed files with 7 additions and 5 deletions
  1. +4
    -4
      mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl
  2. +3
    -1
      mindspore/lite/test/models_tflite_gpu.cfg

+ 4
- 4
mindspore/lite/src/runtime/kernel/opencl/cl/transpose.cl View File

@@ -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));


+ 3
- 1
mindspore/lite/test/models_tflite_gpu.cfg View File

@@ -1,3 +1,5 @@
mobilenet_v1_1.0_224.tflite
mobilenet_v2_1.0_224.tflite
resnet.tflite
resnet.tflite
hiai_cn_recognize_modify_padv2.tflite
hiai_cv_focusShootOCRModel_08.tflite

Loading…
Cancel
Save