Browse Source

mindir 端侧验证成功

tags/v1.6.0
chenping38 chenping 4 years ago
parent
commit
3a65a28895
3 changed files with 6 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c
  2. +3
    -0
      mindspore/lite/test/config/models_mindspore.cfg
  3. +2
    -0
      mindspore/lite/tools/converter/import/primitive_adjust.cc

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/strided_slice_infer.c View File

@@ -441,7 +441,7 @@ int StridedSliceInferShape(const TensorC *const *inputs, size_t inputs_size, Ten
return ret;
}
for (int i = 0; i < transfer_buffer.ndim_; i++) {
if (transfer_buffer.strides_[i] == 0) {
if (transfer_buffer.strides_[i] == 0 || in_shape[i] < transfer_buffer.ends_[i]) {
return NNACL_ERR;
}
output_shape[i] = (transfer_buffer.ends_[i] - transfer_buffer.begins_[i] + transfer_buffer.strides_[i] +


+ 3
- 0
mindspore/lite/test/config/models_mindspore.cfg View File

@@ -8,3 +8,6 @@ mobilenetv2_fzy.mindir 0.5
if_by_if_net.mindir;4 0.01
while_net.mindir;2 0.01
while_in_if_net.mindir;4 0.01
mindspore_FaceRecognitionTrack_1011.mindir
mindspore_FaceQualityAssessment_0928_bs256.mindir
mindspore_nasnet.mindir

+ 2
- 0
mindspore/lite/tools/converter/import/primitive_adjust.cc View File

@@ -225,6 +225,7 @@ int MoveAttrMapCommon(const CNodePtr &cnode) {
}

int MoveAttrMapActivation(const CNodePtr &cnode) {
MS_ASSERT(cnode != nullptr);
auto value_node = cnode->input(0)->cast<ValueNodePtr>();
auto src_prim = GetValueNode<PrimitivePtr>(value_node);
if (src_prim == nullptr) {
@@ -245,6 +246,7 @@ int MoveAttrMapActivation(const CNodePtr &cnode) {
}

int MoveAttrMapActivationGrad(const CNodePtr &cnode) {
MS_ASSERT(cnode != nullptr);
auto value_node = cnode->input(0)->cast<ValueNodePtr>();
auto src_prim = GetValueNode<PrimitivePtr>(value_node);
if (src_prim == nullptr) {


Loading…
Cancel
Save