}
else
{
- // TODO fix indentation and remove clang switch
- // clang-format off
- // test 1.
- // loco node : 1. ConstGen ------------------+-- 4. BiasAdd<Domain::Tensor> -- 5. Push
- // 2. ConstGen - 3. BiasEncode -/
- loco::Graph::NodeContext *loco_nodes = graph->nodes();
-
- loco::Graph::InputContext *loco_inputs = graph->inputs();
- ASSERT_EQ(loco_inputs->size(), 0);
- ASSERT_EQ(loco_nodes->size(), 5);
-
- int idx = 0;
-
- loco::ConstGen *value = dynamic_cast<loco::ConstGen *>(loco_nodes->at(idx++));
- loco::ConstGen *bias = dynamic_cast<loco::ConstGen *>(loco_nodes->at(idx++));
- loco::BiasEncode *enc = dynamic_cast<loco::BiasEncode *>(loco_nodes->at(idx++));
- loco::BiasAdd<loco::Domain::Tensor> *bias_add =
- dynamic_cast<loco::BiasAdd<loco::Domain::Tensor> *>(loco_nodes->at(idx++));
- loco::Push *push = dynamic_cast<loco::Push *>(loco_nodes->at(idx++));
-
- ASSERT_NE(value, nullptr);
- ASSERT_NE(bias, nullptr);
- ASSERT_NE(enc, nullptr);
- ASSERT_NE(bias_add, nullptr);
- ASSERT_NE(push, nullptr);
-
- // check their connection is all OK
- ASSERT_TRUE(enc->input() == bias);
- ASSERT_TRUE(bias_add->bias() == enc);
- ASSERT_TRUE(bias_add->value() == value);
- ASSERT_TRUE(push->from() == bias_add);
-
- // test 2.
- // attrs inside BiasAdd
-
- // axis
- ASSERT_EQ(bias_add->axis(), 3); // NHWC
- // clang-format on
+ // test 1.
+ // loco node : 1. ConstGen ------------------+-- 4. BiasAdd<Domain::Tensor> -- 5. Push
+ // 2. ConstGen - 3. BiasEncode -/
+ loco::Graph::NodeContext *loco_nodes = graph->nodes();
+
+ loco::Graph::InputContext *loco_inputs = graph->inputs();
+ ASSERT_EQ(loco_inputs->size(), 0);
+ ASSERT_EQ(loco_nodes->size(), 5);
+
+ int idx = 0;
+
+ loco::ConstGen *value = dynamic_cast<loco::ConstGen *>(loco_nodes->at(idx++));
+ loco::ConstGen *bias = dynamic_cast<loco::ConstGen *>(loco_nodes->at(idx++));
+ loco::BiasEncode *enc = dynamic_cast<loco::BiasEncode *>(loco_nodes->at(idx++));
+ loco::BiasAdd<loco::Domain::Tensor> *bias_add =
+ dynamic_cast<loco::BiasAdd<loco::Domain::Tensor> *>(loco_nodes->at(idx++));
+ loco::Push *push = dynamic_cast<loco::Push *>(loco_nodes->at(idx++));
+
+ ASSERT_NE(value, nullptr);
+ ASSERT_NE(bias, nullptr);
+ ASSERT_NE(enc, nullptr);
+ ASSERT_NE(bias_add, nullptr);
+ ASSERT_NE(push, nullptr);
+
+ // check their connection is all OK
+ ASSERT_TRUE(enc->input() == bias);
+ ASSERT_TRUE(bias_add->bias() == enc);
+ ASSERT_TRUE(bias_add->value() == value);
+ ASSERT_TRUE(push->from() == bias_add);
+
+ // test 2.
+ // attrs inside BiasAdd
+
+ // axis
+ ASSERT_EQ(bias_add->axis(), 3); // NHWC
}
}
}
else
{
- // TODO fix indentation and remove clang switch
- // clang-format off
-
- // testing axis value of biasAdd
- loco::Graph::NodeContext *loco_nodes = graph->nodes();
- loco::BiasAdd<loco::Domain::Tensor> *bias_add =
- dynamic_cast<loco::BiasAdd<loco::Domain::Tensor> *>(loco_nodes->at(3));
- ASSERT_NE(bias_add, nullptr);
+ // testing axis value of biasAdd
+ loco::Graph::NodeContext *loco_nodes = graph->nodes();
+ loco::BiasAdd<loco::Domain::Tensor> *bias_add =
+ dynamic_cast<loco::BiasAdd<loco::Domain::Tensor> *>(loco_nodes->at(3));
+ ASSERT_NE(bias_add, nullptr);
- ASSERT_EQ(bias_add->axis(), 1); // NCHW
- // clang-format on
+ ASSERT_EQ(bias_add->axis(), 1); // NCHW
}
}