Home
last modified time | relevance | path

Searched refs:matmul_fused_node (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc758 const NodeDef* matmul_fused_node = node_map.GetNode(optimized_name); in TEST_F() local
759 ASSERT_NE(matmul_fused_node, nullptr); in TEST_F()
760 ASSERT_EQ(matmul_fused_node->input_size(), 2); in TEST_F()
761 EXPECT_EQ(matmul_fused_node->input(0), "a"); in TEST_F()
762 EXPECT_EQ(matmul_fused_node->input(1), "b"); in TEST_F()
765 EXPECT_TRUE(matmul_fused_node->attr().at("adj_x").b()); in TEST_F()
766 EXPECT_TRUE(matmul_fused_node->attr().at("adj_y").b()); in TEST_F()
768 EXPECT_TRUE(matmul_fused_node->attr().at("transpose_a").b()); in TEST_F()
769 EXPECT_TRUE(matmul_fused_node->attr().at("transpose_b").b()); in TEST_F()