Function TEST_F

tensorflow/core/common_runtime/optimization_registry_test.cc:81–90

};

TEST_F(OptimizationPassTest, UpdateFuncLibPass) {
  RunPass();
  auto f1 = GetFunctionDef("XTimesTwo");
  ASSERT_NE(f1, nullptr);
  EXPECT_EQ(test::function::XTimesTwo().DebugString(), f1->DebugString());

  auto f2 = GetFunctionDef("WXPlusB");
  ASSERT_NE(f2, nullptr);
  EXPECT_EQ(test::function::WXPlusB().DebugString(), f2->DebugString());
}

}  // namespace tensorflow