Searched refs:CopyNodeAttr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_convolutions.cc | 67 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 …]
|
| D | flatten_atrous.cc | 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()
122 CopyNodeAttr(conv_node, "use_cudnn_on_gpu", "use_cudnn_on_gpu", in FlattenAtrousConv()
|
| D | fold_old_batch_norms.cc | 166 CopyNodeAttr(conv_node, "data_format", "data_format", &bias_add_node); in FuseScaleOffsetToConvWeights()
168 CopyNodeAttr(conv_node, "T", "T", &bias_add_node); in FuseScaleOffsetToConvWeights()
|
| D | transform_utils.h | 69 void CopyNodeAttr(const NodeDef& source, const string& source_key,
|
| D | transform_utils.cc | 146 void CopyNodeAttr(const NodeDef& source, const string& source_key, in CopyNodeAttr() function
|
| D | README.md | 997 CopyNodeAttr(resize_node, "align_corners", "resize_align_corners",
1000 CopyNodeAttr(conv_node, "T", "T", &fused_conv);
1001 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv);
1002 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv);
|
| D | quantize_nodes.cc | 811 CopyNodeAttr(float_node, attr_to_copy, attr_to_copy, in QuantizeNodes()
|
| D | transform_utils_test.cc | 169 CopyNodeAttr(node, "foo", "bar", &copied_node); in TestCopyNodeAttr()
|