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