fix bug for tf

pull/13622/head
xuanyue 4 years ago
parent 1965ecb9a1
commit 051c9c03ba

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

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