!12577 fix bug for dynamic_shape_depends

From: @zhupuxu
Reviewed-by: @jjfeing,@zhoufeng54
Signed-off-by: @zhoufeng54
pull/12577/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 5224241ca7

@ -96,8 +96,6 @@ CNodePtr CreateGatherV2Ds(const FuncGraphPtr &graph, const CNodePtr &origin_node
AnfAlgo::SetOutputInferTypeAndShape({AnfAlgo::GetOutputInferDataType(origin_node, 0)}, {shape}, gather_v2.get());
AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), gather_v2);
AnfAlgo::SetNodeAttr(kAttrInputIsDynamicShape, MakeValue(true), gather_v2);
auto depends_list_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(origin_node, kAttrDynamicShapeDepends);
AnfAlgo::SetNodeAttr(kAttrDynamicShapeDepends, MakeValue(depends_list_me), gather_v2);
auto input_names = AnfAlgo::GetNodeAttr<std::vector<std::string>>(origin_node, kAttrInputNames);
AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), gather_v2);
auto output_names = AnfAlgo::GetNodeAttr<std::vector<std::string>>(origin_node, kAttrOutputNames);

@ -27,7 +27,6 @@
namespace mindspore {
namespace device {
constexpr auto kDynamicShapeDepends = "dynamic_shape_depends";
class DynamicKernel {
public:

@ -368,7 +368,6 @@ constexpr auto kAttrSize = "size";
constexpr auto kAttrIsDynamicShape = "is_dynamic_shape";
constexpr auto kAttrInputIsDynamicShape = "input_is_dynamic_shape";
constexpr auto kAttrOutputIsDynamicShape = "output_is_dynamic_shape";
constexpr auto kAttrDynamicShapeDepends = "dynamic_shape_depends";
constexpr auto kAttrPynativeNextOpName = "next_op";
constexpr auto kAttrPynativeNextIndex = "next_index";
constexpr auto kAttrCompileInfo = "compile_info";

Loading…
Cancel
Save