|
|
|
@@ -740,6 +740,10 @@ schema::PrimitiveT *ZerosLikePrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ZerosLike>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
schema::PrimitiveT *ErfPrimitiveCreator(const AnfNodePtr &node) { |
|
|
|
auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Erf>>(node); |
|
|
|
return ms_primc != nullptr ? ops::MSOp2SchemaOp(ms_primc.get()) : nullptr; |
|
|
|
} |
|
|
|
|
|
|
|
RegistryMSOps g_absPrimitiveCreatorRegistry("Abs", AbsPrimitiveCreator); |
|
|
|
RegistryMSOps g_absGradPrimitiveCreatorRegistry("AbsGrad", AbsGradPrimitiveCreator); |
|
|
|
@@ -949,6 +953,7 @@ RegistryMSOps g_unsortedSegmentSumPrimitiveCreatorRegistry("UnsortedSegmentSum", |
|
|
|
RegistryMSOps g_unsqueezePrimitiveCreatorRegistry("Unsqueeze", UnsqueezePrimitiveCreator); |
|
|
|
RegistryMSOps g_wherePrimitiveCreatorRegistry("Where", WherePrimitiveCreator); |
|
|
|
RegistryMSOps g_zerosLikePrimitiveCreatorRegistry("ZerosLike", ZerosLikePrimitiveCreator); |
|
|
|
RegistryMSOps g_erfPrimitiveCreatorRegistry("Erf", ErfPrimitiveCreator); |
|
|
|
} // namespace lite |
|
|
|
} // namespace mindspore |
|
|
|
|
|
|
|
|