m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix multipe definition of function

Browse Source
pull/7668/head
zhangxuetong 4 years ago
parent 5532de75ef
commit 61178eaaf9
14 changed files with 75 additions and 75 deletions
Show Stats Download Patch File Download Diff File

4
mindspore/lite/nnacl/fp16/activation_fp16.c
Unescape View File

@ -66,7 +66,7 @@ int SigmoidFp16(const float16_t *src, float16_t *dst, int ele_num) {
return NNACL_OK; return NNACL_OK;
} }
float16_t TanhOpt(float16_t src) { float16_t TanhOptFp16(float16_t src) {
if (src > 5.0) { if (src > 5.0) {
return 1.0f; return 1.0f;
} else if (src < -5.0) { } else if (src < -5.0) {
@ -81,7 +81,7 @@ float16_t TanhOpt(float16_t src) {
int TanhFp16(const float16_t *src, float16_t *dst, int ele_num) { int TanhFp16(const float16_t *src, float16_t *dst, int ele_num) {
for (int i = 0; i < ele_num; ++i) { for (int i = 0; i < ele_num; ++i) {
dst[i] = TanhOpt(src[i]); dst[i] = TanhOptFp16(src[i]);
} }
return NNACL_OK; return NNACL_OK;
} }

18
mindspore/lite/nnacl/fp16/crop_fp16.c
Unescape View File

@ -20,25 +20,25 @@
#include "nnacl/crop_parameter.h" #include "nnacl/crop_parameter.h"
void Crop(const float16_t *input, float16_t *output, int task_id, CropParameter *para) { void Fp16Crop(const float16_t *input, float16_t *output, int task_id, CropParameter *para) {
int input_dim = para->input_dim_; int input_dim = para->input_dim_;
switch (input_dim) { switch (input_dim) {
case 1: case 1:
Crop1D(input, output, task_id, para); Fp16Crop1D(input, output, task_id, para);
break; break;
case 2: case 2:
Crop2D(input, output, task_id, para); Fp16Crop2D(input, output, task_id, para);
break; break;
case 3: case 3:
Crop3D(input, output, task_id, para); Fp16Crop3D(input, output, task_id, para);
break; break;
case 4: case 4:
Crop4D(input, output, task_id, para); Fp16Crop4D(input, output, task_id, para);
break; break;
} }
} }
void Crop1D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) { void Fp16Crop1D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) {
const int out_batch = para->out_shape_[0]; const int out_batch = para->out_shape_[0];
const int thread_count = para->thread_count_; const int thread_count = para->thread_count_;
int64_t task_id_stride = thread_count > 1 ? UP_DIV(out_batch, thread_count) : out_batch; int64_t task_id_stride = thread_count > 1 ? UP_DIV(out_batch, thread_count) : out_batch;
@ -55,7 +55,7 @@ void Crop1D(const float16_t *input, float16_t *output, int task_id, CropParamete
memcpy(out_ptr, in_ptr, sizeof(float16_t) * out_dist_stride); memcpy(out_ptr, in_ptr, sizeof(float16_t) * out_dist_stride);
} }
void Crop2D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) { void Fp16Crop2D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) {
const int in_height = para->in_shape_[1]; const int in_height = para->in_shape_[1];
const int out_batch = para->out_shape_[0]; const int out_batch = para->out_shape_[0];
const int out_height = para->out_shape_[1]; const int out_height = para->out_shape_[1];
@ -77,7 +77,7 @@ void Crop2D(const float16_t *input, float16_t *output, int task_id, CropParamete
} }
} }
void Crop3D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) { void Fp16Crop3D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) {
const int in_height = para->in_shape_[1]; const int in_height = para->in_shape_[1];
const int in_width = para->in_shape_[2]; const int in_width = para->in_shape_[2];
@ -111,7 +111,7 @@ void Crop3D(const float16_t *input, float16_t *output, int task_id, CropParamete
} }
} }
void Crop4D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) { void Fp16Crop4D(const float16_t *input, float16_t *output, int task_id, CropParameter *para) {
const int in_height = para->in_shape_[1]; const int in_height = para->in_shape_[1];
const int in_width = para->in_shape_[2]; const int in_width = para->in_shape_[2];
const int in_channel = para->in_shape_[3]; const int in_channel = para->in_shape_[3];

10
mindspore/lite/nnacl/fp16/crop_fp16.h
Unescape View File

@ -24,11 +24,11 @@
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
#endif #endif
void Crop(const float16_t *input, float16_t *output, int task_id, CropParameter *para); void Fp16Crop(const float16_t *input, float16_t *output, int task_id, CropParameter *para);
void Crop1D(const float16_t *input, float16_t *output, int task_id, CropParameter *para); void Fp16Crop1D(const float16_t *input, float16_t *output, int task_id, CropParameter *para);
void Crop2D(const float16_t *input, float16_t *output, int task_id, CropParameter *para); void Fp16Crop2D(const float16_t *input, float16_t *output, int task_id, CropParameter *para);
void Crop3D(const float16_t *input, float16_t *output, int task_id, CropParameter *para); void Fp16Crop3D(const float16_t *input, float16_t *output, int task_id, CropParameter *para);
void Crop4D(const float16_t *input, float16_t *output, int task_id, CropParameter *para); void Fp16Crop4D(const float16_t *input, float16_t *output, int task_id, CropParameter *para);
#ifdef __cplusplus #ifdef __cplusplus
} }
#endif #endif

26
mindspore/lite/nnacl/fp16/scale_fp16.c
Unescape View File

@ -16,7 +16,7 @@
#include "nnacl/fp16/scale_fp16.h" #include "nnacl/fp16/scale_fp16.h"
void ScaleInner(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start, void Fp16ScaleInner(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start,
int outer_end, int axis_size, int inner_size) { int outer_end, int axis_size, int inner_size) {
for (int out = outer_start; out < outer_end; out++) { for (int out = outer_start; out < outer_end; out++) {
int out_offset = out * axis_size * inner_size; int out_offset = out * axis_size * inner_size;
@ -42,7 +42,7 @@ void ScaleInner(float16_t *in_data, float16_t *out_data, float16_t *scale, float
} }
} }
void ScaleAxis(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start, void Fp16ScaleAxis(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start,
int outer_end, int axis_size) { int outer_end, int axis_size) {
for (int out = outer_start; out < outer_end; out++) { for (int out = outer_start; out < outer_end; out++) {
int out_offset = out * axis_size; int out_offset = out * axis_size;
@ -71,14 +71,14 @@ void DoScaleFp16(float16_t *in_data, float16_t *out_data, float16_t *scale, floa
int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_);
if (scale_param->inner_size_ == 1) { if (scale_param->inner_size_ == 1) {
ScaleAxis(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_); Fp16ScaleAxis(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_);
} else { } else {
ScaleInner(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_, Fp16ScaleInner(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_,
scale_param->inner_size_); scale_param->inner_size_);
} }
} }
void ScaleInnerRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start, void Fp16ScaleInnerRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start,
int outer_end, int axis_size, int inner_size) { int outer_end, int axis_size, int inner_size) {
#ifdef ENABLE_ARM64 #ifdef ENABLE_ARM64
float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0}; float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0};
@ -108,7 +108,7 @@ void ScaleInnerRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, f
} }
} }
void ScaleAxisRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start, void Fp16ScaleAxisRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start,
int outer_end, int axis_size) { int outer_end, int axis_size) {
#ifdef ENABLE_ARM64 #ifdef ENABLE_ARM64
float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0}; float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0};
@ -135,21 +135,21 @@ void ScaleAxisRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, fl
} }
} }
void DoScaleReluFp16(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id, void Fp16DoScaleRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id,
ScaleParameter *scale_param) { ScaleParameter *scale_param) {
int outer_step = UP_DIV(scale_param->outer_size_, scale_param->op_parameter_.thread_num_); int outer_step = UP_DIV(scale_param->outer_size_, scale_param->op_parameter_.thread_num_);
int outer_start = task_id * outer_step; int outer_start = task_id * outer_step;
int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_);
if (scale_param->inner_size_ == 1) { if (scale_param->inner_size_ == 1) {
ScaleAxisRelu(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_); Fp16ScaleAxisRelu(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_);
} else { } else {
ScaleInnerRelu(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_, Fp16ScaleInnerRelu(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_,
scale_param->inner_size_); scale_param->inner_size_);
} }
} }
void ScaleInnerRelu6(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start, void Fp16ScaleInnerRelu6(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start,
int outer_end, int axis_size, int inner_size) { int outer_end, int axis_size, int inner_size) {
#ifdef ENABLE_ARM64 #ifdef ENABLE_ARM64
float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0}; float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0};
@ -180,7 +180,7 @@ void ScaleInnerRelu6(float16_t *in_data, float16_t *out_data, float16_t *scale,
} }
} }
void ScaleAxisRelu6(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start, void Fp16ScaleAxisRelu6(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int outer_start,
int outer_end, int axis_size) { int outer_end, int axis_size) {
#ifdef ENABLE_ARM64 #ifdef ENABLE_ARM64
float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0}; float16x8_t zeros = {0, 0, 0, 0, 0, 0, 0, 0};
@ -215,9 +215,9 @@ void DoScaleRelu6Fp16(float16_t *in_data, float16_t *out_data, float16_t *scale,
int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_);
if (scale_param->inner_size_ == 1) { if (scale_param->inner_size_ == 1) {
ScaleAxisRelu6(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_); Fp16ScaleAxisRelu6(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_);
} else { } else {
ScaleInnerRelu6(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_, Fp16ScaleInnerRelu6(in_data, out_data, scale, offset, outer_start, outer_end, scale_param->axis_size_,
scale_param->inner_size_); scale_param->inner_size_);
} }
} }

2
mindspore/lite/nnacl/fp16/scale_fp16.h
Unescape View File

@ -27,7 +27,7 @@ extern "C" {
#endif #endif
void DoScaleFp16(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id, void DoScaleFp16(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id,
ScaleParameter *scale_param); ScaleParameter *scale_param);
void DoScaleReluFp16(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id, void Fp16DoScaleRelu(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id,
ScaleParameter *scale_param); ScaleParameter *scale_param);
void DoScaleRelu6Fp16(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id, void DoScaleRelu6Fp16(float16_t *in_data, float16_t *out_data, float16_t *scale, float16_t *offset, int task_id,
ScaleParameter *scale_param); ScaleParameter *scale_param);

8
mindspore/lite/nnacl/fp16/stack_fp16.c
Unescape View File

@ -17,7 +17,7 @@
#include "nnacl/fp16/stack_fp16.h" #include "nnacl/fp16/stack_fp16.h"
#include "nnacl/arithmetic_common.h" #include "nnacl/arithmetic_common.h"
size_t GetStackCopyNum(int axis, int *in_shape, size_t shape_size) { size_t Fp16GetStackCopyNum(int axis, int *in_shape, size_t shape_size) {
size_t one_input_size = 1; size_t one_input_size = 1;
for (size_t i = 0; i < shape_size; ++i) { for (size_t i = 0; i < shape_size; ++i) {
one_input_size *= in_shape[i]; one_input_size *= in_shape[i];
@ -29,7 +29,7 @@ size_t GetStackCopyNum(int axis, int *in_shape, size_t shape_size) {
return copy_num; return copy_num;
} }
size_t GetStackPreAxisCount(const int *in_shape, int axis) { size_t Fp16GetStackPreAxisCount2(const int *in_shape, int axis) {
size_t pre_axis_count = 1; size_t pre_axis_count = 1;
for (size_t i = 0; i < axis; ++i) { for (size_t i = 0; i < axis; ++i) {
pre_axis_count *= in_shape[i]; pre_axis_count *= in_shape[i];
@ -39,9 +39,9 @@ size_t GetStackPreAxisCount(const int *in_shape, int axis) {
void DoStackFp16(const float16_t *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis, void DoStackFp16(const float16_t *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis,
float16_t *output) { float16_t *output) {
size_t copy_num = GetStackCopyNum(axis, in_shape, shape_size); size_t copy_num = Fp16GetStackCopyNum(axis, in_shape, shape_size);
size_t copy_size = copy_num * sizeof(float16_t); size_t copy_size = copy_num * sizeof(float16_t);
size_t pre_axis_count = GetStackPreAxisCount(in_shape, axis); size_t pre_axis_count = Fp16GetStackPreAxisCount2(in_shape, axis);
size_t in_offset = 0; size_t in_offset = 0;
size_t out_offset = 0; size_t out_offset = 0;
for (size_t i = 0; i < pre_axis_count; ++i) { for (size_t i = 0; i < pre_axis_count; ++i) {

18
mindspore/lite/nnacl/fp16/transpose_fp16.c
Unescape View File

@ -18,7 +18,7 @@
#include <string.h> #include <string.h>
#include "nnacl/errorcode.h" #include "nnacl/errorcode.h"
void TransposeDim2(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm, void Fp16TransposeDim2(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm,
int *output_shape, int h_start, int h_end) { int *output_shape, int h_start, int h_end) {
const int stride0 = strides[perm[0]]; const int stride0 = strides[perm[0]];
const int stride1 = strides[perm[1]]; const int stride1 = strides[perm[1]];
@ -33,7 +33,7 @@ void TransposeDim2(float16_t *in_data, float16_t *out_data, int *strides, int *o
} }
} }
void TransposeDim3(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm, void Fp16TransposeDim3(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm,
int *output_shape, int h_start, int h_end) { int *output_shape, int h_start, int h_end) {
const int stride0 = strides[perm[0]]; const int stride0 = strides[perm[0]];
const int stride1 = strides[perm[1]]; const int stride1 = strides[perm[1]];
@ -56,7 +56,7 @@ void TransposeDim3(float16_t *in_data, float16_t *out_data, int *strides, int *o
} }
} }
void TransposeDim4(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm, void Fp16TransposeDim4(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm,
int *output_shape, int h_start, int h_end) { int *output_shape, int h_start, int h_end) {
const int stride0 = strides[perm[0]]; const int stride0 = strides[perm[0]];
const int stride1 = strides[perm[1]]; const int stride1 = strides[perm[1]];
@ -88,7 +88,7 @@ void TransposeDim4(float16_t *in_data, float16_t *out_data, int *strides, int *o
} }
} }
void TransposeDim5(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm, void Fp16TransposeDim5(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm,
int *output_shape, int h_start, int h_end) { int *output_shape, int h_start, int h_end) {
const int stride0 = strides[perm[0]]; const int stride0 = strides[perm[0]];
const int stride1 = strides[perm[1]]; const int stride1 = strides[perm[1]];
@ -127,7 +127,7 @@ void TransposeDim5(float16_t *in_data, float16_t *out_data, int *strides, int *o
} }
} }
int DoTranspose(float16_t *in_data, float16_t *out_data, int *input_shape, int *output_shape, int Fp16DoTranspose(float16_t *in_data, float16_t *out_data, int *input_shape, int *output_shape,
TransposeParameter *transpose_param, int h_start, int h_end) { TransposeParameter *transpose_param, int h_start, int h_end) {
if (in_data == NULL || out_data == NULL) { if (in_data == NULL || out_data == NULL) {
return NNACL_ERR; return NNACL_ERR;
@ -156,13 +156,13 @@ int DoTranspose(float16_t *in_data, float16_t *out_data, int *input_shape, int *
return NNACL_OK; return NNACL_OK;
} }
if (num_axes == 2) { if (num_axes == 2) {
TransposeDim2(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end); Fp16TransposeDim2(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end);
} else if (num_axes == 3) { } else if (num_axes == 3) {
TransposeDim3(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end); Fp16TransposeDim3(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end);
} else if (num_axes == 4) { } else if (num_axes == 4) {
TransposeDim4(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end); Fp16TransposeDim4(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end);
} else if (num_axes == 5) { } else if (num_axes == 5) {
TransposeDim5(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end); Fp16TransposeDim5(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end);
} }
return NNACL_OK; return NNACL_OK;
} }

2
mindspore/lite/nnacl/fp16/transpose_fp16.h
Unescape View File

@ -35,7 +35,7 @@ typedef struct TransposeParameter {
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
#endif #endif
int DoTranspose(float16_t *in_data, float16_t *out_data, int *input_shape, int *output_shape, int Fp16DoTranspose(float16_t *in_data, float16_t *out_data, int *input_shape, int *output_shape,
TransposeParameter *transpose_param, int h_start, int h_end); TransposeParameter *transpose_param, int h_start, int h_end);
void TransposeDim2(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm, void TransposeDim2(float16_t *in_data, float16_t *out_data, int *strides, int *out_strides, int *perm,
int *output_shape, int h_start, int h_end); int *output_shape, int h_start, int h_end);

14
mindspore/lite/nnacl/int8/crop_int8.c
Unescape View File

@ -18,17 +18,17 @@
#include "nnacl/int8/crop_int8.h" #include "nnacl/int8/crop_int8.h"
#include <string.h> #include <string.h>
void Crop(const int8_t *input, int8_t *output, int task_id, CropParameter *para) { void Int8Crop(const int8_t *input, int8_t *output, int task_id, CropParameter *para) {
int input_dim = para->input_dim_; int input_dim = para->input_dim_;
switch (input_dim) { switch (input_dim) {
case 1: case 1:
Crop1D(input, output, task_id, para); Int8Crop1D(input, output, task_id, para);
break; break;
case 2: case 2:
Crop2D(input, output, task_id, para); Int8Crop2D(input, output, task_id, para);
break; break;
case 3: case 3:
Crop3D(input, output, task_id, para); Int8Crop3D(input, output, task_id, para);
break; break;
case 4: case 4:
Int8Crop4D(input, output, task_id, para); Int8Crop4D(input, output, task_id, para);
@ -36,7 +36,7 @@ void Crop(const int8_t *input, int8_t *output, int task_id, CropParameter *para)
} }
} }
void Crop1D(const int8_t *input, int8_t *output, int task_id, CropParameter *para) { void Int8Crop1D(const int8_t *input, int8_t *output, int task_id, CropParameter *para) {
const int out_batch = para->out_shape_[0]; const int out_batch = para->out_shape_[0];
const int thread_count = para->thread_count_; const int thread_count = para->thread_count_;
int64_t task_id_stride = thread_count > 1 ? UP_DIV(out_batch, thread_count) : out_batch; int64_t task_id_stride = thread_count > 1 ? UP_DIV(out_batch, thread_count) : out_batch;
@ -75,7 +75,7 @@ void Crop1D(const int8_t *input, int8_t *output, int task_id, CropParameter *par
return; return;
} }
void Crop2D(const int8_t *input, int8_t *output, int task_id, CropParameter *para) { void Int8Crop2D(const int8_t *input, int8_t *output, int task_id, CropParameter *para) {
const int in_height = para->in_shape_[1]; const int in_height = para->in_shape_[1];
const int out_batch = para->out_shape_[0]; const int out_batch = para->out_shape_[0];
const int out_height = para->out_shape_[1]; const int out_height = para->out_shape_[1];
@ -118,7 +118,7 @@ void Crop2D(const int8_t *input, int8_t *output, int task_id, CropParameter *par
return; return;
} }
void Crop3D(const int8_t *input, int8_t *output, int task_id, CropParameter *para) { void Int8Crop3D(const int8_t *input, int8_t *output, int task_id, CropParameter *para) {
const int in_height = para->in_shape_[1]; const int in_height = para->in_shape_[1];
const int in_width = para->in_shape_[2]; const int in_width = para->in_shape_[2];

8
mindspore/lite/nnacl/int8/crop_int8.h
Unescape View File

@ -22,10 +22,10 @@
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
#endif #endif
void Crop(const int8_t *input, int8_t *output, int task_id, CropParameter *para); void Int8Crop(const int8_t *input, int8_t *output, int task_id, CropParameter *para);
void Crop1D(const int8_t *input, int8_t *output, int task_id, CropParameter *para); void Int8Crop1D(const int8_t *input, int8_t *output, int task_id, CropParameter *para);
void Crop2D(const int8_t *input, int8_t *output, int task_id, CropParameter *para); void Int8Crop2D(const int8_t *input, int8_t *output, int task_id, CropParameter *para);
void Crop3D(const int8_t *input, int8_t *output, int task_id, CropParameter *para); void Int8Crop3D(const int8_t *input, int8_t *output, int task_id, CropParameter *para);
void Int8Crop4D(const int8_t *input, int8_t *output, int task_id, CropParameter *para); void Int8Crop4D(const int8_t *input, int8_t *output, int task_id, CropParameter *para);
#ifdef __cplusplus #ifdef __cplusplus
} }

2
mindspore/lite/src/runtime/kernel/arm/fp16/crop_fp16.cc
Unescape View File

@ -46,7 +46,7 @@ int CropFp16CPUKernel::Init() {
int CropFp16CPUKernel::ReSize() { return CropBaseCPUKernel::ReSize(); } int CropFp16CPUKernel::ReSize() { return CropBaseCPUKernel::ReSize(); }
int CropFp16CPUKernel::DoExecute(int task_id) { int CropFp16CPUKernel::DoExecute(int task_id) {
Crop(input_ptr_, output_ptr_, task_id, crop_para_); Fp16Crop(input_ptr_, output_ptr_, task_id, crop_para_);
return RET_OK; return RET_OK;
} }

2
mindspore/lite/src/runtime/kernel/arm/fp16/scale_fp16.cc
Unescape View File

@ -80,7 +80,7 @@ int ScaleFp16CPUKernel::Scale(int task_id) {
DoScaleRelu6Fp16(input_, output_, scale_, offset_, task_id, scale_param_); DoScaleRelu6Fp16(input_, output_, scale_, offset_, task_id, scale_param_);
break; break;
case schema::ActivationType_RELU: case schema::ActivationType_RELU:
DoScaleReluFp16(input_, output_, scale_, offset_, task_id, scale_param_); Fp16DoScaleRelu(input_, output_, scale_, offset_, task_id, scale_param_);
break; break;
case schema::ActivationType_NO_ACTIVATION: case schema::ActivationType_NO_ACTIVATION:
DoScaleFp16(input_, output_, scale_, offset_, task_id, scale_param_); DoScaleFp16(input_, output_, scale_, offset_, task_id, scale_param_);

2
mindspore/lite/src/runtime/kernel/arm/fp16/transpose_fp16.cc
Unescape View File

@ -107,7 +107,7 @@ int TransposeFp16CPUKernel::TransposeParallel(int task_id) {
int thread_offset = task_id * thread_h_stride_; int thread_offset = task_id * thread_h_stride_;
TransposeParameter *param = reinterpret_cast<TransposeParameter *>(this->op_parameter_); TransposeParameter *param = reinterpret_cast<TransposeParameter *>(this->op_parameter_);
auto ret = DoTranspose(fp16_in_data_, fp16_out_data_, in_shape_, out_shape_, param, thread_offset, auto ret = Fp16DoTranspose(fp16_in_data_, fp16_out_data_, in_shape_, out_shape_, param, thread_offset,
thread_offset + num_unit_thread); thread_offset + num_unit_thread);
if (ret != RET_OK) { if (ret != RET_OK) {
MS_LOG(ERROR) << "Transpose error task_id[" << task_id << "] error_code[" << ret << "]"; MS_LOG(ERROR) << "Transpose error task_id[" << task_id << "] error_code[" << ret << "]";

2
mindspore/lite/src/runtime/kernel/arm/int8/crop_int8.cc
Unescape View File

@ -84,7 +84,7 @@ int CropInt8CPUKernel::DoExecute(int task_id) {
auto out_tensor = out_tensors_.at(kOutputIndex); auto out_tensor = out_tensors_.at(kOutputIndex);
int8_t *input_data = reinterpret_cast<int8_t *>(input_tensor->MutableData()); int8_t *input_data = reinterpret_cast<int8_t *>(input_tensor->MutableData());
int8_t *output_data = reinterpret_cast<int8_t *>(out_tensor->MutableData()); int8_t *output_data = reinterpret_cast<int8_t *>(out_tensor->MutableData());
Crop(input_data, output_data, task_id, crop_para_); Int8Crop(input_data, output_data, task_id, crop_para_);
return RET_OK; return RET_OK;
} }

Write Preview
Loading…
Cancel
Save