Searched refs:square_node (Results 1 – 1 of 1) sorted by relevance
133 const NodeDef* square_node = node_map.GetNode(absl::StrCat(p, "_", "mul")); in TEST_F() local135 ASSERT_NE(square_node, nullptr); in TEST_F()136 EXPECT_EQ(square_node->op(), "Square"); in TEST_F()137 ASSERT_EQ(square_node->input_size(), 2); in TEST_F()138 EXPECT_EQ(square_node->input(0), "c"); in TEST_F()139 EXPECT_EQ(square_node->input(1), "^d"); in TEST_F()