| @@ -255,7 +255,7 @@ class PynativeEliminater : public OptimizerCaller { | |||||
| MS_LOG(DEBUG) << "Start FillZero"; | MS_LOG(DEBUG) << "Start FillZero"; | ||||
| ValuePtr out = nullptr; | ValuePtr out = nullptr; | ||||
| if (value->isa<Int32Imm>()) { | if (value->isa<Int32Imm>()) { | ||||
| return value; | |||||
| return MakeValue(value->cast<Int32ImmPtr>()->value()); | |||||
| } | } | ||||
| if (value->isa<tensor::Tensor>()) { | if (value->isa<tensor::Tensor>()) { | ||||
| @@ -298,9 +298,10 @@ class PynativeEliminater : public OptimizerCaller { | |||||
| if (rep != nullptr) { | if (rep != nullptr) { | ||||
| if (rep->isa<ValueNode>()) { | if (rep->isa<ValueNode>()) { | ||||
| auto value_node = rep->cast<ValueNodePtr>(); | auto value_node = rep->cast<ValueNodePtr>(); | ||||
| value_node->set_value(FillZero(value_node->value())); | |||||
| auto new_value_node = NewValueNode(FillZero(value_node->value())); | |||||
| new_value_node->set_has_new_value(value_node->has_new_value()); | |||||
| MS_LOG(DEBUG) << "Zeros_like replace ok " << rep->DebugString(4); | MS_LOG(DEBUG) << "Zeros_like replace ok " << rep->DebugString(4); | ||||
| return rep; | |||||
| return new_value_node; | |||||
| } | } | ||||
| } | } | ||||
| } | } | ||||
| @@ -315,9 +316,10 @@ class PynativeEliminater : public OptimizerCaller { | |||||
| if (rep != nullptr) { | if (rep != nullptr) { | ||||
| if (rep->isa<ValueNode>()) { | if (rep->isa<ValueNode>()) { | ||||
| auto value_node = rep->cast<ValueNodePtr>(); | auto value_node = rep->cast<ValueNodePtr>(); | ||||
| value_node->set_value(FillZero(value_node->value())); | |||||
| auto new_value_node = NewValueNode(FillZero(value_node->value())); | |||||
| new_value_node->set_has_new_value(value_node->has_new_value()); | |||||
| MS_LOG(DEBUG) << "Zeros_like replace ok 2 " << rep->DebugString(4); | MS_LOG(DEBUG) << "Zeros_like replace ok 2 " << rep->DebugString(4); | ||||
| return rep; | |||||
| return new_value_node; | |||||
| } | } | ||||
| } | } | ||||
| } | } | ||||