Home
last modified time | relevance | path

Searched refs:shape_input_node (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer_test.cc3100 auto* shape_input_node = context.graph_view->GetNode( in TEST_F() local
3102 ASSERT_NE(shape_input_node, nullptr); in TEST_F()
3103 ASSERT_EQ(shape_input_node->NumRegularFanins(), 2); in TEST_F()
3104 VerifyRegularFaninMatch(shape_input_node, 0, conv2d_transpose_node->GetName(), in TEST_F()