forked from mindspore-Ecosystem/mindspore
!8029 fix static checking of lite ops
Merge pull request !8029 from liuwenhao/master
This commit is contained in:
commit
0f39e5fcdd
|
@ -340,7 +340,7 @@ void DeConvWgMerge(const float *src, float *dst, size_t src_stride, size_t dst_s
|
|||
return;
|
||||
}
|
||||
|
||||
void _deConvWinograd(const float *tile_in, float *tile_out, float *weight_buf, float *tmp_buf, float *at_buf,
|
||||
void _deConvWinograd(const float *tile_in, float *tile_out, float *weight_buf, float *tmp_buf, const float *at_buf,
|
||||
float *a_mid_buf, float *trans_a_buf, bool *transfered, const float *bt_buf, float *b_tmp_buf,
|
||||
int unit_size, int w_start, int h_start, ConvParameter *conv_param, DeConvParam *deconv_param) {
|
||||
int winograd_plane = unit_size * unit_size;
|
||||
|
|
|
@ -79,8 +79,8 @@ void ElementMulAcc(const float *input0, const float *input1, float *output, int
|
|||
}
|
||||
}
|
||||
|
||||
void UpdataState(float *cell_state, const float *forget_gate, const float *input_gate, float *cell_gate, int batch,
|
||||
int hidden_size) {
|
||||
void UpdataState(float *cell_state, const float *forget_gate, const float *input_gate, const float *cell_gate,
|
||||
int batch, int hidden_size) {
|
||||
ElementMul(forget_gate, cell_state, cell_state, batch * hidden_size);
|
||||
ElementMulAcc(input_gate, cell_gate, cell_state, batch * hidden_size);
|
||||
}
|
||||
|
|
|
@ -65,7 +65,7 @@ int PrepareResizeBilinear(const int *input_shape, const int *output_shape, bool
|
|||
}
|
||||
|
||||
int ResizeBilinear(const float *input_data, float *output_data, const int *input_shape, const int *output_shape,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, float *y_bottom_weights,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, const float *y_bottom_weights,
|
||||
float *x_left_weights, int n_h_begin, int n_h_end) {
|
||||
if (input_data == NULL || output_data == NULL || input_shape == NULL || output_shape == NULL || y_bottoms == NULL ||
|
||||
y_tops == NULL || x_lefts == NULL || x_rights == NULL || y_bottom_weights == NULL || x_left_weights == NULL) {
|
||||
|
@ -154,8 +154,8 @@ int ResizeBilinear(const float *input_data, float *output_data, const int *input
|
|||
return NNACL_OK;
|
||||
}
|
||||
|
||||
int InterpRow(const float *src_line, float *linear_output, int new_width, float *x_left_weights, const int *x_lefts,
|
||||
const int *x_rights, int in_c) {
|
||||
int InterpRow(const float *src_line, float *linear_output, int new_width, const float *x_left_weights,
|
||||
const int *x_lefts, const int *x_rights, int in_c) {
|
||||
int w;
|
||||
for (w = 0; w < new_width; w++) {
|
||||
int c = 0;
|
||||
|
@ -208,7 +208,7 @@ int InterpCol(const float *bottom_line, const float *top_line, float *output, in
|
|||
}
|
||||
|
||||
int ResizeBilinear2(const float *input_data, float *output_data, const int *input_shape, const int *output_shape,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, float *y_bottom_weights,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, const float *y_bottom_weights,
|
||||
float *x_left_weights, float *line0, float *line1, int n_h_begin, int n_h_end) {
|
||||
if (input_data == NULL || output_data == NULL || input_shape == NULL || output_shape == NULL || y_bottoms == NULL ||
|
||||
y_tops == NULL || x_lefts == NULL || x_rights == NULL || y_bottom_weights == NULL || x_left_weights == NULL) {
|
||||
|
|
|
@ -30,11 +30,11 @@ int PrepareResizeBilinear(const int *input_shape, const int *output_shape, bool
|
|||
int *y_tops, int *x_lefts, int *x_rights, float *y_bottom_weights, float *x_left_weights);
|
||||
|
||||
int ResizeBilinear(const float *input_data, float *output_data, const int *input_shape, const int *output_shape,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, float *y_bottom_weights,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, const float *y_bottom_weights,
|
||||
float *x_left_weights, int n_h_begin, int n_h_end);
|
||||
|
||||
int ResizeBilinear2(const float *input_data, float *output_data, const int *input_shape, const int *output_shape,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, float *y_bottom_weights,
|
||||
const int *y_bottoms, const int *y_tops, int *x_lefts, int *x_rights, const float *y_bottom_weights,
|
||||
float *x_left_weights, float *line0, float *line1, int n_h_begin, int n_h_end);
|
||||
|
||||
int ResizeNearestNeighbor(const float *input_data, float *output_data, const int *input_shape, const int *output_shape,
|
||||
|
|
|
@ -18,7 +18,7 @@
|
|||
#ifdef ENABLE_ARM
|
||||
#include <arm_neon.h>
|
||||
#endif
|
||||
void ScaleInner(float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
void ScaleInner(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
int outer_end, int axis_size, int inner_size) {
|
||||
for (int out = outer_start; out < outer_end; out++) {
|
||||
int out_offset = out * axis_size * inner_size;
|
||||
|
@ -43,8 +43,8 @@ void ScaleInner(float *in_data, float *out_data, const float *scale, const float
|
|||
}
|
||||
}
|
||||
|
||||
void ScaleAxis(float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, int outer_end,
|
||||
int axis_size) {
|
||||
void ScaleAxis(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
int outer_end, int axis_size) {
|
||||
for (int out = outer_start; out < outer_end; out++) {
|
||||
int out_offset = out * axis_size;
|
||||
int index = 0;
|
||||
|
@ -65,7 +65,7 @@ void ScaleAxis(float *in_data, float *out_data, const float *scale, const float
|
|||
}
|
||||
}
|
||||
|
||||
void DoScale(float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
void DoScale(const float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
ScaleParameter *scale_param) {
|
||||
int outer_step = UP_DIV(scale_param->outer_size_, scale_param->op_parameter_.thread_num_);
|
||||
int outer_start = task_id * outer_step;
|
||||
|
@ -79,7 +79,7 @@ void DoScale(float *in_data, float *out_data, const float *scale, float *offset,
|
|||
}
|
||||
}
|
||||
|
||||
void ScaleInnerRelu(float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
void ScaleInnerRelu(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
int outer_end, int axis_size, int inner_size) {
|
||||
#ifdef ENABLE_ARM64
|
||||
float32x4_t zeros = {0, 0, 0, 0};
|
||||
|
@ -109,7 +109,7 @@ void ScaleInnerRelu(float *in_data, float *out_data, const float *scale, const f
|
|||
}
|
||||
}
|
||||
|
||||
void ScaleAxisRelu(float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
void ScaleAxisRelu(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
int outer_end, int axis_size) {
|
||||
#ifdef ENABLE_ARM64
|
||||
float32x4_t zeros = {0, 0, 0, 0};
|
||||
|
@ -136,7 +136,7 @@ void ScaleAxisRelu(float *in_data, float *out_data, const float *scale, const fl
|
|||
}
|
||||
}
|
||||
|
||||
void DoScaleRelu(float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
void DoScaleRelu(const float *in_data, float *out_data, const float *scale, const float *offset, int task_id,
|
||||
ScaleParameter *scale_param) {
|
||||
int outer_step = UP_DIV(scale_param->outer_size_, scale_param->op_parameter_.thread_num_);
|
||||
int outer_start = task_id * outer_step;
|
||||
|
@ -150,7 +150,7 @@ void DoScaleRelu(float *in_data, float *out_data, const float *scale, float *off
|
|||
}
|
||||
}
|
||||
|
||||
void ScaleInnerRelu6(float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
void ScaleInnerRelu6(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
int outer_end, int axis_size, int inner_size) {
|
||||
#ifdef ENABLE_ARM64
|
||||
float32x4_t zeros = {0, 0, 0, 0};
|
||||
|
@ -181,7 +181,7 @@ void ScaleInnerRelu6(float *in_data, float *out_data, const float *scale, const
|
|||
}
|
||||
}
|
||||
|
||||
void ScaleAxisRelu6(float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
void ScaleAxisRelu6(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start,
|
||||
int outer_end, int axis_size) {
|
||||
#ifdef ENABLE_ARM64
|
||||
float32x4_t zeros = {0, 0, 0, 0};
|
||||
|
@ -209,7 +209,7 @@ void ScaleAxisRelu6(float *in_data, float *out_data, const float *scale, const f
|
|||
}
|
||||
}
|
||||
|
||||
void DoScaleRelu6(float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
void DoScaleRelu6(const float *in_data, float *out_data, const float *scale, const float *offset, int task_id,
|
||||
ScaleParameter *scale_param) {
|
||||
int outer_step = UP_DIV(scale_param->outer_size_, scale_param->op_parameter_.thread_num_);
|
||||
int outer_start = task_id * outer_step;
|
||||
|
|
|
@ -22,11 +22,11 @@
|
|||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
void DoScale(float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
void DoScale(const float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
ScaleParameter *scale_param);
|
||||
void DoScaleRelu(float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
void DoScaleRelu(const float *in_data, float *out_data, const float *scale, const float *offset, int task_id,
|
||||
ScaleParameter *scale_param);
|
||||
void DoScaleRelu6(float *in_data, float *out_data, const float *scale, float *offset, int task_id,
|
||||
void DoScaleRelu6(const float *in_data, float *out_data, const float *scale, const float *offset, int task_id,
|
||||
ScaleParameter *scale_param);
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
|
|
|
@ -16,7 +16,7 @@
|
|||
#include "nnacl/fp32/space_to_batch.h"
|
||||
#include "nnacl/arithmetic_common.h"
|
||||
|
||||
void DoSpaceToBatchNHWC(const float *input, float *output, const int *block_sizes, int *in_shape,
|
||||
void DoSpaceToBatchNHWC(const float *input, float *output, const int *block_sizes, const int *in_shape,
|
||||
const int *out_shape) {
|
||||
int out_dim0 = out_shape[0];
|
||||
int out_dim1 = out_shape[1];
|
||||
|
|
|
@ -30,7 +30,8 @@ typedef struct SpaceToBatchParameter {
|
|||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
void DoSpaceToBatchNHWC(const float *input, float *output, const int *block_sizes, int *in_shape, const int *out_shape);
|
||||
void DoSpaceToBatchNHWC(const float *input, float *output, const int *block_sizes, const int *in_shape,
|
||||
const int *out_shape);
|
||||
void DoSpaceToBatchPaddingNHWC(const float *input, float *output, int *in_shape, const int *padding,
|
||||
const int *out_shape);
|
||||
#ifdef __cplusplus
|
||||
|
|
|
@ -18,7 +18,7 @@
|
|||
#include "nnacl/errorcode.h"
|
||||
#include "nnacl/op_base.h"
|
||||
|
||||
int SpaceToDepthForNHWC(const float *input, float *output, int *in_shape, const int *out_shape, int shape_size,
|
||||
int SpaceToDepthForNHWC(const float *input, float *output, const int *in_shape, const int *out_shape, int shape_size,
|
||||
int block_size, int h_start, int h_end) {
|
||||
if (input == NULL || output == NULL) {
|
||||
return NNACL_NULL_PTR;
|
||||
|
|
|
@ -24,7 +24,7 @@ typedef struct SpaceToDepthParameter {
|
|||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
int SpaceToDepthForNHWC(const float *input, float *output, int *in_shape, const int *out_shape, int shape_size,
|
||||
int SpaceToDepthForNHWC(const float *input, float *output, const int *in_shape, const int *out_shape, int shape_size,
|
||||
int block_size, int h_start, int h_end);
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
|
|
|
@ -37,7 +37,8 @@ size_t GetStackPreAxisCount(const int *in_shape, int axis) {
|
|||
return pre_axis_count;
|
||||
}
|
||||
|
||||
void DoStack(const float *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis, float *output) {
|
||||
void DoStack(const float *const *inputs, size_t input_num, const int *in_shape, size_t shape_size, int axis,
|
||||
float *output) {
|
||||
size_t copy_num = GetStackCopyNum(axis, in_shape, shape_size);
|
||||
size_t copy_size = copy_num * sizeof(float);
|
||||
size_t pre_axis_count = GetStackPreAxisCount(in_shape, axis);
|
||||
|
@ -52,7 +53,7 @@ void DoStack(const float *const *inputs, size_t input_num, int *in_shape, size_t
|
|||
}
|
||||
}
|
||||
|
||||
void DoStackInt32(const int32_t *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis,
|
||||
void DoStackInt32(const int32_t *const *inputs, size_t input_num, const int *in_shape, size_t shape_size, int axis,
|
||||
int32_t *output) {
|
||||
size_t copy_num = GetStackCopyNum(axis, in_shape, shape_size);
|
||||
size_t copy_size = copy_num * sizeof(int32_t);
|
||||
|
|
|
@ -21,8 +21,9 @@
|
|||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
void DoStack(const float *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis, float *output);
|
||||
void DoStackInt32(const int32_t *const *inputs, size_t input_num, int *in_shape, size_t shape_size, int axis,
|
||||
void DoStack(const float *const *inputs, size_t input_num, const int *in_shape, size_t shape_size, int axis,
|
||||
float *output);
|
||||
void DoStackInt32(const int32_t *const *inputs, size_t input_num, const int *in_shape, size_t shape_size, int axis,
|
||||
int32_t *output);
|
||||
void DoStackOneInput(const int8_t *input, int8_t *output, size_t data_size);
|
||||
#ifdef __cplusplus
|
||||
|
|
|
@ -491,7 +491,7 @@ void ConvDw3x3Int8Pad(int8_t *output_data, const int8_t *input_data, const int16
|
|||
/*conv depthwise sliding window perchannel int8 begin*/
|
||||
void DepthwiseBorderPixelInt8(int8_t *dst, const int8_t *src, const int16_t *weight, const int32_t *bias, int height,
|
||||
int width, int in_kh_step, int in_kw_step, int kernel_w, int8_t *input_zp,
|
||||
int32_t *out_zp, const int *out_multiplier, int *left_shift, const int *right_shift,
|
||||
int32_t *out_zp, const int *out_multiplier, const int *left_shift, const int *right_shift,
|
||||
int32_t *acc_min, int32_t *acc_max) {
|
||||
int tmp_buffer[C8NUM];
|
||||
for (int i = 0; i < C8NUM; i++) {
|
||||
|
@ -527,7 +527,7 @@ void DepthwiseBorderPixelInt8(int8_t *dst, const int8_t *src, const int16_t *wei
|
|||
|
||||
void DepthwiseBorderInt8(int8_t *dst, const int8_t *src, const int16_t *weight, const int32_t *bias, int top,
|
||||
int bottom, int left, int right, const ConvParameter *conv_param,
|
||||
const SlidingWindowParam *sliding, int8_t *in_zp, int32_t *out_zp, int *out_multiplier,
|
||||
const SlidingWindowParam *sliding, int8_t *in_zp, int32_t *out_zp, const int *out_multiplier,
|
||||
int *left_shift, const int *right_shift, int32_t *acc_min, int32_t *acc_max) {
|
||||
int8_t *dst_h = dst + top * sliding->out_h_step_;
|
||||
for (int oh = top; oh < bottom; oh++) {
|
||||
|
|
|
@ -16,7 +16,7 @@
|
|||
#include "nnacl/int8/space_to_batch_int8.h"
|
||||
#include "nnacl/arithmetic_common.h"
|
||||
|
||||
void DoSpaceToBatchNHWCInt8(const int8_t *input, int8_t *output, const int *block_sizes, int *in_shape,
|
||||
void DoSpaceToBatchNHWCInt8(const int8_t *input, int8_t *output, const int *block_sizes, const int *in_shape,
|
||||
const int *out_shape) {
|
||||
int out_dim0 = out_shape[0];
|
||||
int out_dim1 = out_shape[1];
|
||||
|
@ -46,7 +46,7 @@ void DoSpaceToBatchNHWCInt8(const int8_t *input, int8_t *output, const int *bloc
|
|||
}
|
||||
}
|
||||
|
||||
void DoSpaceToBatchPaddingNHWCInt8(const int8_t *input, int8_t *output, int *in_shape, const int *padding,
|
||||
void DoSpaceToBatchPaddingNHWCInt8(const int8_t *input, int8_t *output, const int *in_shape, const int *padding,
|
||||
const int *out_shape, int32_t zp) {
|
||||
int in_h = in_shape[1];
|
||||
int in_w = in_shape[2];
|
||||
|
|
|
@ -21,9 +21,9 @@
|
|||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
void DoSpaceToBatchNHWCInt8(const int8_t *input, int8_t *output, const int *block_sizes, int *in_shape,
|
||||
void DoSpaceToBatchNHWCInt8(const int8_t *input, int8_t *output, const int *block_sizes, const int *in_shape,
|
||||
const int *out_shape);
|
||||
void DoSpaceToBatchPaddingNHWCInt8(const int8_t *input, int8_t *output, int *in_shape, const int *padding,
|
||||
void DoSpaceToBatchPaddingNHWCInt8(const int8_t *input, int8_t *output, const int *in_shape, const int *padding,
|
||||
const int *out_shape, int32_t zp);
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
|
|
|
@ -103,7 +103,7 @@ void T(const float *poly_array, float *matrix_t, int n) {
|
|||
}
|
||||
}
|
||||
|
||||
int B(float *poly_array, float *matrix_b, int in_unit) {
|
||||
int B(const float *poly_array, float *matrix_b, int in_unit) {
|
||||
memset(matrix_b, 0, in_unit * in_unit * sizeof(float));
|
||||
int n = in_unit - 1;
|
||||
if ((n * n) > MAX_LEN || (n * in_unit) > MAX_LEN) {
|
||||
|
|
|
@ -36,7 +36,7 @@ int LT(const float *poly_array, float *matrix_lt, int n);
|
|||
|
||||
void T(const float *poly_array, float *matrix_t, int n);
|
||||
|
||||
int B(float *poly_array, float *matrix_b, int in_unit);
|
||||
int B(const float *poly_array, float *matrix_b, int in_unit);
|
||||
|
||||
void GenerateIntervalArray(float *array, float interval, int degree);
|
||||
|
||||
|
|
|
@ -19,7 +19,7 @@
|
|||
#include "nnacl/errorcode.h"
|
||||
|
||||
void TransposeDim2(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end) {
|
||||
const int *output_shape, int h_start, int h_end) {
|
||||
const int stride0 = strides[perm[0]];
|
||||
const int stride1 = strides[perm[1]];
|
||||
const int output0 = output_shape[0];
|
||||
|
@ -34,7 +34,7 @@ void TransposeDim2(float *in_data, float *out_data, const int *strides, int *out
|
|||
}
|
||||
|
||||
void TransposeDim3(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end) {
|
||||
const int *output_shape, int h_start, int h_end) {
|
||||
const int stride0 = strides[perm[0]];
|
||||
const int stride1 = strides[perm[1]];
|
||||
const int stride2 = strides[perm[2]];
|
||||
|
@ -57,7 +57,7 @@ void TransposeDim3(float *in_data, float *out_data, const int *strides, int *out
|
|||
}
|
||||
|
||||
void TransposeDim4(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end) {
|
||||
const int *output_shape, int h_start, int h_end) {
|
||||
const int stride0 = strides[perm[0]];
|
||||
const int stride1 = strides[perm[1]];
|
||||
const int stride2 = strides[perm[2]];
|
||||
|
@ -89,7 +89,7 @@ void TransposeDim4(float *in_data, float *out_data, const int *strides, int *out
|
|||
}
|
||||
|
||||
void TransposeDim5(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end) {
|
||||
const int *output_shape, int h_start, int h_end) {
|
||||
const int stride0 = strides[perm[0]];
|
||||
const int stride1 = strides[perm[1]];
|
||||
const int stride2 = strides[perm[2]];
|
||||
|
@ -128,7 +128,7 @@ void TransposeDim5(float *in_data, float *out_data, const int *strides, int *out
|
|||
}
|
||||
|
||||
void TransposeDims(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end, int dims, int *size, int *position) {
|
||||
const int *output_shape, int h_start, int h_end, int dims, int *size, int *position) {
|
||||
*(size + dims - 1) = 1;
|
||||
for (int i = dims - 1; i > 0; --i) {
|
||||
*(size + i - 1) = *(size + i) * output_shape[i];
|
||||
|
|
|
@ -35,15 +35,15 @@ extern "C" {
|
|||
int DoTranspose(float *in_data, float *out_data, int *input_shape, int *output_shape,
|
||||
TransposeParameter *transpose_param, int h_start, int h_end, int *size, int *position);
|
||||
void TransposeDim2(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end);
|
||||
const int *output_shape, int h_start, int h_end);
|
||||
void TransposeDim3(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end);
|
||||
const int *output_shape, int h_start, int h_end);
|
||||
void TransposeDim4(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end);
|
||||
const int *output_shape, int h_start, int h_end);
|
||||
void TransposeDim5(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end);
|
||||
const int *output_shape, int h_start, int h_end);
|
||||
void TransposeDims(float *in_data, float *out_data, const int *strides, int *out_strides, const int *perm,
|
||||
int *output_shape, int h_start, int h_end, int dims, int *size, int *position);
|
||||
const int *output_shape, int h_start, int h_end, int dims, int *size, int *position);
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
|
|
@ -40,6 +40,7 @@ int BinaryCrossEntropy::UnPackAttr(const Primitive &prim, const std::vector<AnfN
|
|||
<< schema::EnumNamePrimitiveType(primitive_->value.type) << "is not equal"
|
||||
<< schema::EnumNamePrimitiveType(schema::PrimitiveType_BinaryCrossEntropy);
|
||||
delete this->primitive_;
|
||||
this->primitive_ = nullptr;
|
||||
return RET_ERROR;
|
||||
}
|
||||
if (this->primitive_->value.value == nullptr) {
|
||||
|
@ -47,6 +48,7 @@ int BinaryCrossEntropy::UnPackAttr(const Primitive &prim, const std::vector<AnfN
|
|||
if (attr == nullptr) {
|
||||
MS_LOG(ERROR) << "new binary cross entropy attr failed!";
|
||||
delete this->primitive_;
|
||||
this->primitive_ = nullptr;
|
||||
return RET_ERROR;
|
||||
}
|
||||
// default is mean
|
||||
|
@ -55,6 +57,8 @@ int BinaryCrossEntropy::UnPackAttr(const Primitive &prim, const std::vector<AnfN
|
|||
MS_LOG(ERROR) << "get reduction failed!";
|
||||
delete this->primitive_;
|
||||
delete attr;
|
||||
this->primitive_ = nullptr;
|
||||
attr = nullptr;
|
||||
return RET_ERROR;
|
||||
} else {
|
||||
reduction = GetValue<string>(prim.GetAttr("reduction"));
|
||||
|
|
|
@ -88,14 +88,12 @@ void DeConvolutionWinogradCPUKernel::FreeDeconvParam() {
|
|||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (deconv_param_->compute_units_ != nullptr) {
|
||||
free(deconv_param_->compute_units_);
|
||||
deconv_param_->compute_units_ = nullptr;
|
||||
}
|
||||
if (deconv_param_->compute_units_ != nullptr) {
|
||||
free(deconv_param_->compute_units_);
|
||||
deconv_param_->compute_units_ = nullptr;
|
||||
}
|
||||
|
||||
if (deconv_param_ != nullptr) {
|
||||
delete (deconv_param_);
|
||||
deconv_param_ = nullptr;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue