|
|
|
@@ -193,7 +193,7 @@ inline const PrimitivePtr kPrimSoftmax = std::make_shared<Primitive>("Softmax"); |
|
|
|
inline const PrimitivePtr kPrimSparseSoftmaxCrossEntropy = std::make_shared<Primitive>("SparseSoftmaxCrossEntropy"); |
|
|
|
inline const PrimitivePtr kPrimLogSoftmax = std::make_shared<Primitive>("LogSoftmax"); |
|
|
|
inline const PrimitivePtr kPrimLogSoftmaxGrad = std::make_shared<Primitive>("LogSoftmaxGrad"); |
|
|
|
inline const PrimitivePtr kPrimLstm = std::make_shared<Primitive>("Lstm"); |
|
|
|
inline const PrimitivePtr kPrimLstm = std::make_shared<Primitive>("LSTM"); |
|
|
|
inline const PrimitivePtr kPrimTan = std::make_shared<Primitive>("Tan"); |
|
|
|
inline const PrimitivePtr kPrimAtan2 = std::make_shared<Primitive>("Atan2"); |
|
|
|
inline const PrimitivePtr kPrimAtan = std::make_shared<Primitive>("Atan"); |
|
|
|
|