Home
last modified time | relevance | path

Searched refs:CopyNodeAttr (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions.cc67 CopyNodeAttr(resize_node, "align_corners", "resize_align_corners", in FuseResizePadAndConv()
69 CopyNodeAttr(mirror_pad_node, "mode", "mode", &fused_conv); in FuseResizePadAndConv()
70 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizePadAndConv()
71 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv); in FuseResizePadAndConv()
72 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv); in FuseResizePadAndConv()
122 CopyNodeAttr(resize_node, "align_corners", "resize_align_corners", in FuseResizeAndConv()
125 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizeAndConv()
126 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv); in FuseResizeAndConv()
127 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv); in FuseResizeAndConv()
178 CopyNodeAttr(mirror_pad_node, "mode", "mode", &fused_conv); in FusePadAndConv()
[all …]
Dflatten_atrous.cc115 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()
122 CopyNodeAttr(conv_node, "use_cudnn_on_gpu", "use_cudnn_on_gpu", in FlattenAtrousConv()
Dfold_old_batch_norms.cc166 CopyNodeAttr(conv_node, "data_format", "data_format", &bias_add_node); in FuseScaleOffsetToConvWeights()
168 CopyNodeAttr(conv_node, "T", "T", &bias_add_node); in FuseScaleOffsetToConvWeights()
Dtransform_utils.h69 void CopyNodeAttr(const NodeDef& source, const string& source_key,
Dtransform_utils.cc146 void CopyNodeAttr(const NodeDef& source, const string& source_key, in CopyNodeAttr() function
DREADME.md995 CopyNodeAttr(resize_node, "align_corners", "resize_align_corners",
998 CopyNodeAttr(conv_node, "T", "T", &fused_conv);
999 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv);
1000 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv);
Dquantize_nodes.cc811 CopyNodeAttr(float_node, attr_to_copy, attr_to_copy, in QuantizeNodes()
Dtransform_utils_test.cc169 CopyNodeAttr(node, "foo", "bar", &copied_node); in TestCopyNodeAttr()