|
|
|
@@ -25,7 +25,6 @@ namespace mindspore { |
|
|
|
namespace opt { |
|
|
|
ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() { |
|
|
|
Register(prim::kPrimCast->name(), {1}); |
|
|
|
Register(prim::kPrimAvgPoolGrad->name(), {0}); |
|
|
|
Register(prim::kPrimAvgPoolGradVm->name(), {0}); |
|
|
|
Register(prim::kPrimConv2DBackpropInput->name(), {2}); |
|
|
|
Register(prim::kPrimConv2DBackpropFilter->name(), {2}); |
|
|
|
|