Searched refs:input_transpose_node (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 414 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 …]
|
| D | generic_layout_optimizer_test.cc | 247 auto* input_transpose_node = graph_view.GetNode( in TEST_F() local
251 ASSERT_NE(input_transpose_node, nullptr); in TEST_F()
252 ASSERT_EQ(input_transpose_node->NumRegularFanins(), 2); in TEST_F()
253 VerifyRegularFaninMatch(input_transpose_node, 0, "Input", 0); in TEST_F()
258 VerifyRegularFaninMatch(conv2d_node, 0, input_transpose_node->GetName(), 0); in TEST_F()
|