Searched refs:input_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 | 901 auto* input_transpose_node2 = context.graph_view->GetNode( in TEST_F() local
903 ASSERT_NE(input_transpose_node2, nullptr); in TEST_F()
904 ASSERT_EQ(input_transpose_node2->NumRegularFanins(), 2); in TEST_F()
905 VerifyRegularFaninMatch(input_transpose_node2, 0, "orig_output", 0); in TEST_F()
919 input_transpose_node2->GetName(), 0); in TEST_F()
1032 auto* input_transpose_node2 = context.graph_view->GetNode( in TEST_F() local
1034 ASSERT_NE(input_transpose_node2, nullptr); in TEST_F()
1035 ASSERT_EQ(input_transpose_node2->NumRegularFanins(), 2); in TEST_F()
1036 VerifyRegularFaninMatch(input_transpose_node2, 0, "out_backprop", 0); in TEST_F()
1043 VerifyRegularFaninMatch(conv2d_bf_node, 2, input_transpose_node2->GetName(), in TEST_F()
[all …]
|