Searched refs:CopyNodeAttr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_convolutions.cc | 68 CopyNodeAttr(resize_node, "align_corners", "resize_align_corners", in FuseResizePadAndConv()
70 CopyNodeAttr(mirror_pad_node, "mode", "mode", &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()
123 CopyNodeAttr(resize_node, "align_corners", "resize_align_corners", in FuseResizeAndConv()
126 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizeAndConv()
127 CopyNodeAttr(conv_node, "padding", "padding", &fused_conv); in FuseResizeAndConv()
128 CopyNodeAttr(conv_node, "strides", "strides", &fused_conv); in FuseResizeAndConv()
179 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 | fuse_quantized_convolution.cc | 117 CopyNodeAttr(quantized_conv2D_node, "Tinput", "Tinput", &fused_conv); in FuseQuantizedConvolutionAndRequantize()
118 CopyNodeAttr(quantized_conv2D_node, "Tfilter", "Tfilter", &fused_conv); in FuseQuantizedConvolutionAndRequantize()
119 CopyNodeAttr(quantized_conv2D_node, "strides", "strides", &fused_conv); in FuseQuantizedConvolutionAndRequantize()
120 CopyNodeAttr(quantized_conv2D_node, "padding", "padding", &fused_conv); in FuseQuantizedConvolutionAndRequantize()
124 CopyNodeAttr(quantized_conv2D_node, "dilations", in FuseQuantizedConvolutionAndRequantize()
|
| D | fold_old_batch_norms.cc | 167 CopyNodeAttr(conv_node, "data_format", "data_format", &bias_add_node); in FuseScaleOffsetToConvWeights()
169 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 | 144 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 | 806 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()
|