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.cc827 const NodeDef* matmul_fused_node = node_map.GetNode(optimized_name); in TEST_F() local
828 ASSERT_NE(matmul_fused_node, nullptr); in TEST_F()
829 EXPECT_EQ("a", matmul_fused_node->input(0)); in TEST_F()
830 EXPECT_EQ("b", matmul_fused_node->input(1)); in TEST_F()
833 EXPECT_TRUE(matmul_fused_node->attr().at("adj_x").b()); in TEST_F()
834 EXPECT_TRUE(matmul_fused_node->attr().at("adj_y").b()); in TEST_F()
836 EXPECT_TRUE(matmul_fused_node->attr().at("transpose_a").b()); in TEST_F()
837 EXPECT_TRUE(matmul_fused_node->attr().at("transpose_b").b()); in TEST_F()