Browse Source

!13162 [MS_LITE] print log

From: @YeFeng_24
Reviewed-by: @hangangqiang,@zhanghaibo5,@hangangqiang
Signed-off-by: @hangangqiang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
1c9da602a4
5 changed files with 9 additions and 4 deletions
  1. +1
    -1
      mindspore/lite/nnacl/infer/random_standard_normal_infer.c
  2. +5
    -0
      mindspore/lite/src/ops/ops_utils.cc
  3. +1
    -1
      mindspore/lite/tools/converter/legacy_optimizer/graph/format_trans_pass.cc
  4. +1
    -1
      mindspore/lite/tools/converter/parser/tf/tf_model_parser.cc
  5. +1
    -1
      mindspore/lite/tools/optimizer/graph/infershape_pass.cc

+ 1
- 1
mindspore/lite/nnacl/infer/random_standard_normal_infer.c View File

@@ -41,6 +41,6 @@ int RandomStandardNormalInferShape(const TensorC *const *inputs, size_t inputs_s
}
SetShapeArray(outputs[0], output_shape, output_shape_size);
outputs[0]->data_type_ = kNumberTypeFloat32;
outputs[0]->format_ = Format_NHWC;
outputs[0]->format_ = inputs[0]->format_;
return NNACL_OK;
}

+ 5
- 0
mindspore/lite/src/ops/ops_utils.cc View File

@@ -742,6 +742,10 @@ schema::PrimitiveT *ZerosLikePrimitiveCreator(const AnfNodePtr &node) {
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ZerosLike>>(node);
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr;
}
schema::PrimitiveT *WhilePrimitiveCreator(const AnfNodePtr &node) {
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::While>>(node);
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr;
}

RegistryMSOps g_absPrimitiveCreatorRegistry("Abs", AbsPrimitiveCreator);
RegistryMSOps g_absGradPrimitiveCreatorRegistry("AbsGrad", AbsGradPrimitiveCreator);
@@ -952,6 +956,7 @@ RegistryMSOps g_unsortedSegmentSumPrimitiveCreatorRegistry("UnsortedSegmentSum",
RegistryMSOps g_unsqueezePrimitiveCreatorRegistry("Unsqueeze", UnsqueezePrimitiveCreator);
RegistryMSOps g_wherePrimitiveCreatorRegistry("Where", WherePrimitiveCreator);
RegistryMSOps g_zerosLikePrimitiveCreatorRegistry("ZerosLike", ZerosLikePrimitiveCreator);
RegistryMSOps g_whilePrimitiveCreatorRegistry("While", WhilePrimitiveCreator);
} // namespace lite
} // namespace mindspore



+ 1
- 1
mindspore/lite/tools/converter/legacy_optimizer/graph/format_trans_pass.cc View File

@@ -282,7 +282,7 @@ STATUS FormatTransPass::ChangeOpAxis(schema::MetaGraphT *graph, const std::uniqu
return RET_NOT_SUPPORT;
}
} else {
MS_LOG(ERROR) << "change op axis only support 4 dims";
MS_LOG(DEBUG) << "change op axis only support 4 dims";
return RET_NOT_SUPPORT;
}
}


+ 1
- 1
mindspore/lite/tools/converter/parser/tf/tf_model_parser.cc View File

@@ -184,7 +184,7 @@ STATUS TFModelParser::ConvertConstVariant(const tensorflow::TensorProto &tensor_
}
auto &variant = tensor_proto.variant_val(0);
if (variant.type_name() != "tensorflow::TensorList" || variant.tensors_size() <= 0) {
MS_LOG(ERROR) << "Only nonempty TensorList type is supported now";
MS_LOG(DEBUG) << "Only nonempty TensorList type is supported now";
}
auto descriptor = variant.GetMetadata().descriptor;
auto reflection = variant.GetMetadata().reflection;


+ 1
- 1
mindspore/lite/tools/optimizer/graph/infershape_pass.cc View File

@@ -388,7 +388,7 @@ bool InferShapePass::Run(const FuncGraphPtr &func_graph) {
}
auto prim_t = lite::GetPrimitiveT(cnode->input(0));
if (prim_t == nullptr) {
MS_LOG(ERROR) << "prim_t is nullptr";
MS_LOG(DEBUG) << "prim_t is nullptr";
FreeTensors(&input_tensors);
FreeTensors(&output_tensors);
return false;


Loading…
Cancel
Save