浏览代码

GPU CI: add OCR02 and OCR08

tags/v1.0.0
chenzupeng 5 年前
父节点
当前提交
24d6f7db31
共有 2 个文件被更改,包括 7 次插入5 次删除
  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 查看文件

@@ -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 查看文件

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

正在加载...
取消
保存