diff --git a/src/common/types.cpp b/src/common/types.cpp index b90fb70f2a81b365f049632cc7281a69ec58e18d..ede49478ce1a56ca603f5f8580d4d0c231d6616d 100644 --- a/src/common/types.cpp +++ b/src/common/types.cpp @@ -82,6 +82,7 @@ std::unordered_map< {G_OP_TYPE_FUSION_CONV_ADD, {{"Input"}, {"Out"}}}, {G_OP_TYPE_RELU, {{"X"}, {"Out"}}}, {G_OP_TYPE_SOFTMAX, {{"X"}, {"Out"}}}, + {G_OP_TYPE_SIGMOID, {{"X"}, {"Out"}}}, {G_OP_TYPE_MUL, {{"X"}, {"Out"}}}, {G_OP_TYPE_ELEMENTWISE_ADD, {{"X", "Y"}, {"Out"}}}, {G_OP_TYPE_POOL2D, {{"X"}, {"Out"}}},