Function TEST

compiler/loco/src/Service/CanonicalShapeInferenceRule.test.cpp:26–44

#include <gtest/gtest.h>

TEST(CanonicalShapeInferenceRuleTest, minimal)
{
  // Create a simple identity network, which takes Tensor<1x2x3x4> as input.
  GraphTestcase<GraphCode::Identity> testcase{1, 2, 3, 4};

  // Run Inference
  loco::CanonicalShapeInferenceRule rule;

  loco::apply(&rule).to(testcase.graph());

  // Verify!
  ASSERT_TRUE(loco::shape_known(testcase.push_node));
  ASSERT_EQ(loco::Domain::Tensor, loco::shape_get(testcase.push_node).domain());
  ASSERT_EQ(4, loco::shape_get(testcase.push_node).as<loco::TensorShape>().rank());
  ASSERT_EQ(1, loco::shape_get(testcase.push_node).as<loco::TensorShape>().dim(0));
  ASSERT_EQ(2, loco::shape_get(testcase.push_node).as<loco::TensorShape>().dim(1));
  ASSERT_EQ(3, loco::shape_get(testcase.push_node).as<loco::TensorShape>().dim(2));
  ASSERT_EQ(4, loco::shape_get(testcase.push_node).as<loco::TensorShape>().dim(3));
}

TEST(CanonicalShapeInferenceRuleTest, const_gen)
{