Browse Source

!13622 [lite]fix bug for tf

From: @xu_anyue
Reviewed-by: @hangangqiang,@jpc_chenjianping
Signed-off-by: @hangangqiang
pull/13622/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
72f64291ec
2 changed files with 2 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/tools/converter/parser/tf/tf_arithmetic_parser.cc
  2. +1
    -0
      mindspore/lite/tools/converter/parser/tf/tf_batch_matmul_parser.cc

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

@@ -404,7 +404,7 @@ ops::PrimitiveC *TFPowParser::Parse(const tensorflow::NodeDef &tf_op,
auto prim = std::make_unique<ops::PowFusion>();

*output_size = 1;
if (AddOpInput(tf_op, 0, inputs) != RET_OK) {
if (AddOpInput(tf_op, 0, inputs) != RET_OK || AddOpInput(tf_op, 1, inputs) != RET_OK) {
MS_LOG(ERROR) << "Add Op input failed.";
return nullptr;
}


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

@@ -47,5 +47,6 @@ ops::PrimitiveC *TFBatchMatMulParser::Parse(const tensorflow::NodeDef &tf_op,
return prim.release();
}
TFNodeRegistrar g_tfBatchMatMulParser("BatchMatMul", new TFBatchMatMulParser());
TFNodeRegistrar g_tfBatchMatMulV2Parser("BatchMatMulV2", new TFBatchMatMulParser());
} // namespace lite
} // namespace mindspore

Loading…
Cancel
Save