Searched refs:AddNodeInput (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | quantize_nodes.cc | 363 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 …]
|
D | fold_old_batch_norms_test.cc | 92 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 …]
|
D | fuse_convolutions.cc | 64 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 …]
|
D | fuse_quantized_convolution.cc | 82 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()
|
D | sparsify_gather.cc | 418 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()
|
D | quantize_weights.cc | 131 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()
|
D | flatten_atrous.cc | 112 AddNodeInput(input_node.name(), &flattened_conv_node); in FlattenAtrousConv() 113 AddNodeInput(upsampled_filter_node.name(), &flattened_conv_node); in FlattenAtrousConv()
|
D | fold_old_batch_norms.cc | 170 AddNodeInput(conv_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights() 171 AddNodeInput(bias_offset_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights()
|
D | transform_utils.h | 66 void AddNodeInput(const string& input_name, NodeDef* node);
|
D | transform_utils_test.cc | 159 AddNodeInput("foo", &node); in TestAddNodeInput() 862 AddNodeInput("b", &a); in TestCopyOriginalMatch()
|
D | transform_utils.cc | 140 void AddNodeInput(const string& input_name, NodeDef* node) { in AddNodeInput() function
|
D | README.md | 993 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);
|