diff --git a/mace/ops/lstm_nonlinear.cc b/mace/ops/lstm_nonlinear.cc index 596c9ad77bd3add3d99043e5a5f4ebd33db5dade..bf28c79e43f79307e074922f0c94ebf311901f25 100644 --- a/mace/ops/lstm_nonlinear.cc +++ b/mace/ops/lstm_nonlinear.cc @@ -101,7 +101,7 @@ class LSTMNonlinearOp : public Operation { }; void RegisterLSTMNonlinear(OpRegistryBase *op_registry) { - MACE_REGISTER_OP(op_registry, "LSTMNonlinearOp", LSTMNonlinearOp, + MACE_REGISTER_OP(op_registry, "LSTMNonlinear", LSTMNonlinearOp, DeviceType::CPU, float); }