Searched refs:square_node (Results 1 – 1 of 1) sorted by relevance
235 const NodeDef* square_node = node_map.GetNode(strings::StrCat(p, "_", "mul")); in TEST_F() local237 ASSERT_NE(square_node, nullptr); in TEST_F()238 EXPECT_EQ("Square", square_node->op()); in TEST_F()239 EXPECT_EQ("c", square_node->input(0)); in TEST_F()240 EXPECT_EQ("^d", square_node->input(1)); in TEST_F()