Searched refs:output_transpose_node2 (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | generic_layout_optimizer_transposer_test.cc | 1562 auto* output_transpose_node2 = context.graph_view->GetNode( in TEST_F() local 1564 ASSERT_NE(output_transpose_node2, nullptr); in TEST_F() 1565 ASSERT_EQ(output_transpose_node2->NumRegularFanins(), 2); in TEST_F() 1566 VerifyRegularFaninMatch(output_transpose_node2, 0, in_node->GetName(), 1); in TEST_F() 1585 output_transpose_node2->GetName(), 0); in TEST_F() 1820 auto* output_transpose_node2 = context.graph_view->GetNode( in TEST_F() local 1822 ASSERT_NE(output_transpose_node2, nullptr); in TEST_F() 1823 ASSERT_EQ(output_transpose_node2->NumRegularFanins(), 2); in TEST_F() 1824 VerifyRegularFaninMatch(output_transpose_node2, 0, switch_node->GetName(), 1); in TEST_F() 1834 VerifyRegularFaninMatch(i2_node, 0, output_transpose_node2->GetName(), 0); in TEST_F()
|