From 1c0df41db00636d1b40a588b92325e19293ca773 Mon Sep 17 00:00:00 2001 From: liutuo Date: Fri, 26 Apr 2019 10:00:29 +0800 Subject: [PATCH] fix lstm_nonlinear name --- mace/ops/lstm_nonlinear.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/ops/lstm_nonlinear.cc b/mace/ops/lstm_nonlinear.cc index 596c9ad7..bf28c79e 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); } -- GitLab