Searched refs:folded_conv (Results 1 – 1 of 1) sorted by relevance
1579 const NodeDef* folded_conv = node_map.GetNode(conv.node()->name()); in TEST_F() local1580 ASSERT_NE(folded_conv, nullptr); in TEST_F()1583 node_map.GetNode(folded_conv->input(1)); in TEST_F()1589 node_map.GetNode(NodeName(folded_conv->input(0))); in TEST_F()1663 const NodeDef* folded_conv = node_map.GetNode(conv.node()->name()); in TEST_F() local1664 ASSERT_NE(folded_conv, nullptr); in TEST_F()1665 ASSERT_EQ(folded_conv->input_size(), 2); in TEST_F()1666 CHECK_EQ(NodeName(folded_conv->input(0)), inputs.node()->name()); in TEST_F()1668 node_map.GetNode(NodeName(folded_conv->input(1))); in TEST_F()