Home
last modified time | relevance | path

Searched refs:AddNodeInput (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc363 AddNodeInput(node.name() + rename_suffix, &dequantize_node); in QuantizePlaceholders()
364 AddNodeInput(min_node.name(), &dequantize_node); in QuantizePlaceholders()
365 AddNodeInput(max_node.name(), &dequantize_node); in QuantizePlaceholders()
422 AddNodeInput(fake_quant_node.input(0), &quantize_node); in ConvertFakeQuantsToRequantize()
423 AddNodeInput(fake_quant_min_node.name(), &quantize_node); in ConvertFakeQuantsToRequantize()
424 AddNodeInput(fake_quant_max_node.name(), &quantize_node); in ConvertFakeQuantsToRequantize()
432 AddNodeInput(quantize_node.name() + ":0", &requantize_node); in ConvertFakeQuantsToRequantize()
433 AddNodeInput(quantize_node.name() + ":1", &requantize_node); in ConvertFakeQuantsToRequantize()
434 AddNodeInput(quantize_node.name() + ":2", &requantize_node); in ConvertFakeQuantsToRequantize()
435 AddNodeInput(fake_quant_min_node.name(), &requantize_node); in ConvertFakeQuantsToRequantize()
[all …]
Dfold_old_batch_norms_test.cc92 AddNodeInput("conv_op", &batch_norm_node); in TestFoldOldBatchNorms()
93 AddNodeInput("mean_op", &batch_norm_node); in TestFoldOldBatchNorms()
94 AddNodeInput("variance_op", &batch_norm_node); in TestFoldOldBatchNorms()
95 AddNodeInput("beta_op", &batch_norm_node); in TestFoldOldBatchNorms()
96 AddNodeInput("gamma_op", &batch_norm_node); in TestFoldOldBatchNorms()
170 AddNodeInput("conv_op", &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
171 AddNodeInput("mean_op", &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
172 AddNodeInput("variance_op", &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
173 AddNodeInput("beta_op", &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
174 AddNodeInput("gamma_op", &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
[all …]
Dfuse_convolutions.cc64 AddNodeInput(resize_node.input(0), &fused_conv); in FuseResizePadAndConv()
65 AddNodeInput(resize_node.input(1), &fused_conv); in FuseResizePadAndConv()
66 AddNodeInput(mirror_pad_node.input(1), &fused_conv); in FuseResizePadAndConv()
67 AddNodeInput(conv_node.input(1), &fused_conv); in FuseResizePadAndConv()
119 AddNodeInput(resize_node.input(0), &fused_conv); in FuseResizeAndConv()
120 AddNodeInput(resize_node.input(1), &fused_conv); in FuseResizeAndConv()
121 AddNodeInput(pad_dims_node.name(), &fused_conv); in FuseResizeAndConv()
122 AddNodeInput(conv_node.input(1), &fused_conv); in FuseResizeAndConv()
176 AddNodeInput(mirror_pad_node.input(0), &fused_conv); in FusePadAndConv()
177 AddNodeInput(mirror_pad_node.input(1), &fused_conv); in FusePadAndConv()
[all …]
Dfuse_quantized_convolution.cc82 AddNodeInput(quantized_conv2D_node.input(i), &fused_conv); in FuseQuantizedConvolutionAndRequantize()
84 AddNodeInput(const_requantize_range_min_node.name(), &fused_conv); in FuseQuantizedConvolutionAndRequantize()
85 AddNodeInput(const_requantize_range_max_node.name(), &fused_conv); in FuseQuantizedConvolutionAndRequantize()
96 AddNodeInput(summand, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
97 AddNodeInput(min_summand, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
98 AddNodeInput(max_summand, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
Dsparsify_gather.cc418 AddNodeInput(hashtable_node.name(), &init_table_node); in SparsifyGatherInternal()
420 AddNodeInput(indices_node.name(), &init_table_node); in SparsifyGatherInternal()
422 AddNodeInput(values_node.name(), &init_table_node); in SparsifyGatherInternal()
425 AddNodeInput(hashtable_node.name(), &lookup_node); in SparsifyGatherInternal()
427 AddNodeInput(gather_node.input(1), &lookup_node); in SparsifyGatherInternal()
429 AddNodeInput(default_value_node.name(), &lookup_node); in SparsifyGatherInternal()
432 AddNodeInput(lookup_node.name(), &expand_dims_node); in SparsifyGatherInternal()
434 AddNodeInput(dim_idx_node.name(), &expand_dims_node); in SparsifyGatherInternal()
468 AddNodeInput(StrCat("^", name), init_op); in SparsifyGatherInternal()
Dquantize_weights.cc131 AddNodeInput(quantized_const_node.name(), &dequantize_node); in QuantizeWeights()
132 AddNodeInput(min_node.name(), &dequantize_node); in QuantizeWeights()
133 AddNodeInput(max_node.name(), &dequantize_node); in QuantizeWeights()
Dflatten_atrous.cc112 AddNodeInput(input_node.name(), &flattened_conv_node); in FlattenAtrousConv()
113 AddNodeInput(upsampled_filter_node.name(), &flattened_conv_node); in FlattenAtrousConv()
Dfold_old_batch_norms.cc170 AddNodeInput(conv_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights()
171 AddNodeInput(bias_offset_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights()
Dtransform_utils.h66 void AddNodeInput(const string& input_name, NodeDef* node);
Dtransform_utils_test.cc159 AddNodeInput("foo", &node); in TestAddNodeInput()
862 AddNodeInput("b", &a); in TestCopyOriginalMatch()
Dtransform_utils.cc140 void AddNodeInput(const string& input_name, NodeDef* node) { in AddNodeInput() function
DREADME.md993 AddNodeInput(resize_node.input(0), &fused_conv);
994 AddNodeInput(resize_node.input(1), &fused_conv);
995 AddNodeInput(pad_dims_node.name(), &fused_conv);
996 AddNodeInput(conv_node.input(1), &fused_conv);