|
|
|
@ -60,9 +60,9 @@ TEST(OpKernel, all) {
|
|
|
|
|
|
|
|
|
|
TEST(NetOp, insert_op) {
|
|
|
|
|
NetOp net;
|
|
|
|
|
auto op1 = std::shared_ptr<NOP>(
|
|
|
|
|
new NOP("empty", {{"X", {"x"}}, {"W", {"w1"}}, {"b", {"b1"}}},
|
|
|
|
|
{{"Out", {"y"}}}, {}));
|
|
|
|
|
auto op1 = std::shared_ptr<framework::NOP>(
|
|
|
|
|
new framework::NOP("empty", {{"X", {"x"}}, {"W", {"w1"}}, {"b", {"b1"}}},
|
|
|
|
|
{{"Out", {"y"}}}, {}));
|
|
|
|
|
net.AddOp(op1);
|
|
|
|
|
net.InsertOp(0, op1);
|
|
|
|
|
ASSERT_EQ(2UL, net.ops_.size());
|
|
|
|
|