Searched refs:input_transpose_node1 (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 | 895 auto* input_transpose_node1 = context.graph_view->GetNode( in TEST_F() local
897 ASSERT_NE(input_transpose_node1, nullptr); in TEST_F()
898 ASSERT_EQ(input_transpose_node1->NumRegularFanins(), 2); in TEST_F()
899 VerifyRegularFaninMatch(input_transpose_node1, 0, "orig_input", 0); in TEST_F()
917 input_transpose_node1->GetName(), 0); in TEST_F()
1022 auto* input_transpose_node1 = context.graph_view->GetNode( in TEST_F() local
1024 ASSERT_NE(input_transpose_node1, nullptr); in TEST_F()
1025 ASSERT_EQ(input_transpose_node1->NumRegularFanins(), 2); in TEST_F()
1026 VerifyRegularFaninMatch(input_transpose_node1, 0, "input", 0); in TEST_F()
1041 VerifyRegularFaninMatch(conv2d_bf_node, 0, input_transpose_node1->GetName(), in TEST_F()
[all …]
|