|
|
|
@ -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");
|
|
|
|
|