Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc255 auto* conv2d_node = graph_view.GetNode("Conv2D"); in TEST_F() local
256 ASSERT_NE(conv2d_node, nullptr); in TEST_F()
257 ASSERT_EQ(conv2d_node->NumRegularFanins(), 2); in TEST_F()
258 VerifyRegularFaninMatch(conv2d_node, 0, input_transpose_node->GetName(), 0); in TEST_F()
259 VerifyRegularFaninMatch(conv2d_node, 1, "Filter", 0); in TEST_F()
260 VerifyDataFormatAttributeMatch(conv2d_node, DST_DATA_FORMAT); in TEST_F()
267 VerifyRegularFaninMatch(output_transpose_node, 0, conv2d_node->GetName(), 0); in TEST_F()
476 auto* conv2d_node = graph_view.GetNode("Conv2D"); in TEST_F() local
477 ASSERT_NE(conv2d_node, nullptr); in TEST_F()
478 ASSERT_EQ(conv2d_node->NumRegularFanins(), 2); in TEST_F()
[all …]
Dgeneric_layout_optimizer_transposer_test.cc852 auto* conv2d_node = context.graph_view->GetNode("conv2d"); in TEST_F() local
853 ASSERT_NE(conv2d_node, nullptr); in TEST_F()
854 ASSERT_EQ(conv2d_node->NumRegularFanins(), 2); in TEST_F()
855 VerifyRegularFaninMatch(conv2d_node, 0, input_transpose_node->GetName(), 0); in TEST_F()
856 VerifyRegularFaninMatch(conv2d_node, 1, "filter", 0); in TEST_F()
857 VerifyDataFormatAttributeMatch(conv2d_node, kDstFormat); in TEST_F()
858 const auto* strides_attr = conv2d_node->GetAttr("strides"); in TEST_F()
870 VerifyRegularFaninMatch(output_transpose_node, 0, conv2d_node->GetName(), 0); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2225 Node* conv2d_node = nullptr; in SetUpContiguousInputs() local
2250 if (conv2d_node != nullptr) { in SetUpContiguousInputs()
2259 conv2d_node = e->dst(); in SetUpContiguousInputs()
2290 nb->Input(conv2d_node, kConv2DFilterOutputSlotIdx); in SetUpContiguousInputs()
2332 GetNodeProducingMklTensor(g, old_node, conv2d_node, in SetUpContiguousInputs()