Home
last modified time | relevance | path

Searched refs:input_transpose_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.cc414 auto* input_transpose_node = context.graph_view->GetNode( in ReduceTransposerKeepDims() local
416 ASSERT_NE(input_transpose_node, nullptr); in ReduceTransposerKeepDims()
422 input_transpose_node->GetName(), 0); in ReduceTransposerKeepDims()
481 auto* input_transpose_node = context.graph_view->GetNode( in ReduceTransposerValidAxisNode() local
483 ASSERT_NE(input_transpose_node, nullptr); in ReduceTransposerValidAxisNode()
489 input_transpose_node->GetName(), 0); in ReduceTransposerValidAxisNode()
789 auto* input_transpose_node = in TEST_F() local
791 ASSERT_NE(input_transpose_node, nullptr); in TEST_F()
792 ASSERT_EQ(input_transpose_node->NumRegularFanins(), 2); in TEST_F()
793 VerifyRegularFaninMatch(input_transpose_node, 0, "x", 0); in TEST_F()
[all …]