xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug of resize

This commit is contained in:
wang_shaocong 2020-11-30 10:44:31 +08:00
parent 0c153b586a
commit c80dd2dba3
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/nnacl/int8/resize_int8.c
View File

@ -93,7 +93,7 @@ int ResizeBilinearWithFloatScaleInt8(const int8_t *input_ptr, int8_t *output_ptr
int input_rt_index = (y_upper_value * in_w + x_upper_value) * channel;
int c = 0;
#ifdef ENABLE_ARM
for (; c < channel; c += 4) {
for (; c <= channel - 4; c += 4) {
float32x4_t in_lb;
in_lb[0] = (float)in_b_ptr[input_lb_index];
in_lb[1] = (float)in_b_ptr[input_lb_index + 1];

4
mindspore/lite/src/ops/populate/strided_slice_populate.h
View File

@ -14,6 +14,9 @@
* limitations under the License.
*/
#ifndef LITE_MINDSPORE_LITE_C_OPS_OP_STRIDED_SLICE_POPULATE_H
#define LITE_MINDSPORE_LITE_C_OPS_OP_STRIDED_SLICE_POPULATE_H
#include "src/ops/arithmetic.h"
namespace mindspore {
@ -23,3 +26,4 @@ OpParameter *PopulateStridedSliceParameter(const mindspore::lite::PrimitiveC *pr
} // namespace lite
} // namespace mindspore
#endif