Searched refs:output_transpose_node (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 | 430 auto* output_transpose_node = context.graph_view->GetNode( in ReduceTransposerKeepDims() local
432 ASSERT_NE(output_transpose_node, nullptr); in ReduceTransposerKeepDims()
437 VerifyRegularFaninMatch(z_output_node, 0, output_transpose_node->GetName(), in ReduceTransposerKeepDims()
805 auto* output_transpose_node = in TEST_F() local
807 ASSERT_NE(output_transpose_node, nullptr); in TEST_F()
808 ASSERT_EQ(output_transpose_node->NumRegularFanins(), 2); in TEST_F()
809 VerifyRegularFaninMatch(output_transpose_node, 0, bn_node->GetName(), 0); in TEST_F()
814 VerifyRegularFaninMatch(output_y, 0, output_transpose_node->GetName(), 0); in TEST_F()
866 auto* output_transpose_node = context.graph_view->GetNode( in TEST_F() local
868 ASSERT_NE(output_transpose_node, nullptr); in TEST_F()
[all …]
|