!8728 code review problem fix

From: @zhaozhenlong
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2020-11-19 09:39:09 +08:00 committed by Gitee
commit 567338924f
4 changed files with 20 additions and 3 deletions

View File

@ -16,10 +16,13 @@
#include "nnacl/fp32/reverse_fp32.h"
#include <string.h>
#include "nnacl/op_base.h"
#include "nnacl/errorcode.h"
#include "nnacl/nnacl_utils.h"
int Reverse(const float *input, float *output, size_t elem_size, int *index) {
for (int i = 0; i < elem_size; i++) {
NNACL_ASSERT(index[i] >= 0);
}
for (int i = 0; i < elem_size; i++) {
output[index[i]] = input[i];
}

View File

@ -176,6 +176,11 @@ int DoTransposeFp32(const float *in_data, float *out_data, int *input_shape, con
(void)memcpy(out_data, in_data, data_size);
return NNACL_OK;
}
for (int i = 0; i < num_axes; ++i) {
if (perm[i] < 0) {
return NNACL_PARAM_INVALID;
}
}
if (num_axes == 2) {
TransposeDim2Fp32(in_data, out_data, strides, out_strides, perm, output_shape, h_start, h_end);
} else if (num_axes == 3) {

View File

@ -21,9 +21,18 @@
#ifdef __cplusplus
extern "C" {
#endif
#if defined(__arm__) || defined(__aarch64__)
uint32_t getHwCap(int hwcap_type);
#endif
#ifdef DEBUG
#include <assert.h>
#define NNACL_ASSERT(f) assert(f)
#else
#define NNACL_ASSERT(f) ((void)0)
#endif
#ifdef __cplusplus
}
#endif

View File

@ -20,7 +20,7 @@
int DoSqueeze(const float *in_data, float *out_data, size_t data_size) {
if (in_data == NULL || out_data == NULL) {
return -1;
return NNACL_ERR;
}
(void)memcpy(out_data, in_data, data_size);
return NNACL_OK;
@ -28,7 +28,7 @@ int DoSqueeze(const float *in_data, float *out_data, size_t data_size) {
int DoSqueezeInt32(int32_t *in_data, int32_t *out_data, size_t data_size) {
if (in_data == NULL || out_data == NULL) {
return -1;
return NNACL_ERR;
}
(void)memcpy(out_data, in_data, data_size);
return NNACL_OK;