fix bug of resize

pull/9198/head
wang_shaocong 4 years ago
parent 0c153b586a
commit c80dd2dba3

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

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

Loading…
Cancel
Save