| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_convolutions.cc | 50 const NodeDef& conv_node = match.node; in FuseResizePadAndConv() local
67 AddNodeInput(conv_node.input(1), &fused_conv); in FuseResizePadAndConv()
71 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizePadAndConv()
72 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv); in FuseResizePadAndConv()
73 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv); in FuseResizePadAndConv()
99 const NodeDef& conv_node = match.node; in FuseResizeAndConv() local
109 pad_dims_node.set_name(conv_node.name() + "_dummy_paddings"); in FuseResizeAndConv()
122 AddNodeInput(conv_node.input(1), &fused_conv); in FuseResizeAndConv()
126 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizeAndConv()
127 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv); in FuseResizeAndConv()
[all …]
|
| D | flatten_atrous.cc | 55 const NodeDef& conv_node = match.inputs[0].node; in FlattenAtrousConv() local
109 flattened_conv_node.set_op(conv_node.op()); in FlattenAtrousConv()
110 flattened_conv_node.set_device(conv_node.device()); in FlattenAtrousConv()
115 CopyNodeAttr(conv_node, "T", "T", &flattened_conv_node); in FlattenAtrousConv()
116 CopyNodeAttr(conv_node, "strides", "strides", &flattened_conv_node); in FlattenAtrousConv()
118 CopyNodeAttr(conv_node, "data_format", "data_format", in FlattenAtrousConv()
121 if (conv_node.op() == "Conv2D") { in FlattenAtrousConv()
122 CopyNodeAttr(conv_node, "use_cudnn_on_gpu", "use_cudnn_on_gpu", in FlattenAtrousConv()
|
| D | fold_batch_norms.cc | 54 const NodeDef& conv_node = match.inputs[0].node; in FoldBatchNorms() local
60 for (const auto& node : {conv_node, weights_node, mul_values_node}) { in FoldBatchNorms()
64 {mul_node, conv_node, input_node, weights_node, in FoldBatchNorms()
76 if (conv_node.op() == "Conv2D") { in FoldBatchNorms()
78 } else if (conv_node.op() == "DepthwiseConv2dNative") { in FoldBatchNorms()
112 new_conv_node = conv_node; in FoldBatchNorms()
|
| D | fold_old_batch_norms.cc | 111 const NodeDef& conv_node = conv_node_match.node; in FuseScaleOffsetToConvWeights() local
119 if (conv_node.op() == "Conv2D") { in FuseScaleOffsetToConvWeights()
121 } else if (conv_node.op() == "DepthwiseConv2dNative") { in FuseScaleOffsetToConvWeights()
154 new_nodes->push_back(conv_node); in FuseScaleOffsetToConvWeights()
158 bias_offset_node.set_name(conv_node.name() + "_bn_offset"); in FuseScaleOffsetToConvWeights()
166 if (!conv_node.attr().count("data_format")) { in FuseScaleOffsetToConvWeights()
167 CopyNodeAttr(conv_node, "data_format", "data_format", &bias_add_node); in FuseScaleOffsetToConvWeights()
169 CopyNodeAttr(conv_node, "T", "T", &bias_add_node); in FuseScaleOffsetToConvWeights()
170 AddNodeInput(conv_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights()
205 const NodeDef& conv_node = conv_node_match.node; in FuseBatchNormWithBatchToSpace() local
[all …]
|
| D | README.md | 973 const NodeDef& conv_node = match.node;
983 pad_dims_node.set_name(conv_node.name() + "_dummy_paddings");
996 AddNodeInput(conv_node.input(1), &fused_conv);
1000 CopyNodeAttr(conv_node, "T", "T", &fused_conv);
1001 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv);
1002 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 460 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
461 EXPECT_EQ(conv_node->attr().at({"data_format"}).s(), "NHWC"); in TEST_F()
474 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
475 EXPECT_EQ(conv_node->attr().at({"data_format"}).s(), "NCHW"); in TEST_F()
489 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
490 EXPECT_EQ(conv_node->attr().at({"data_format"}).s(), "NCHW"); in TEST_F()
504 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
505 EXPECT_EQ(conv_node->attr().at({"data_format"}).s(), "NHWC"); in TEST_F()
518 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
519 EXPECT_EQ(conv_node->attr().at({"data_format"}).s(), "NHWC"); in TEST_F()
[all …]
|
| D | constant_folding.cc | 2915 NodeDef* conv_node = in MulConvPushDown() local
2917 if (!IsConv2D(*conv_node) && !IsConv3D(*conv_node)) { in MulConvPushDown()
2927 if (conv_node->input_size() < 2 || in MulConvPushDown()
2928 NumNonControlOutputs(*conv_node, *node_map_) > 1 || in MulConvPushDown()
2929 nodes_to_preserve_.find(conv_node->name()) != nodes_to_preserve_.end()) { in MulConvPushDown()
2934 NodeDef* conv_left_child = node_map_->GetNode(conv_node->input(0)); in MulConvPushDown()
2935 NodeDef* conv_right_child = node_map_->GetNode(conv_node->input(1)); in MulConvPushDown()
2947 const auto& conv_props = properties.GetOutputProperties(conv_node->name()); in MulConvPushDown()
2957 const auto& input_props = properties.GetInputProperties(conv_node->name()); in MulConvPushDown()
2971 conv_node->attr().at("data_format").s(), filter_shape, const_shape)) { in MulConvPushDown()
[all …]
|
| D | arithmetic_optimizer_test.cc | 1691 const NodeDef* conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
1697 ASSERT_NE(conv_node, nullptr); in TEST_F()
1702 EXPECT_EQ(conv_node->input(0), cast_node->name()); in TEST_F()
1703 EXPECT_EQ(conv_node->input(1), weights_node->name()); in TEST_F()
1742 const NodeDef* conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
1747 ASSERT_NE(conv_node, nullptr); in TEST_F()
1750 EXPECT_EQ(conv_node->input(1), weights_node->name()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 910 const Node* conv_node; in GetPadOrConv2D() local
916 conv_node = n; in GetPadOrConv2D()
927 conv_node = m; in GetPadOrConv2D()
936 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrConv2D()
959 const Node* conv_node; in GetPadOrFusedConv2D() local
966 conv_node = n; in GetPadOrFusedConv2D()
977 conv_node = m; in GetPadOrFusedConv2D()
985 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrFusedConv2D()
|