Home
last modified time | relevance | path

Searched refs:SetNodeAttr (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc336 SetNodeAttr("dtype", DT_QUINT8, &quantized_placeholder); in QuantizePlaceholders()
342 SetNodeAttr("dtype", DT_FLOAT, &min_node); in QuantizePlaceholders()
351 SetNodeAttr("dtype", DT_FLOAT, &max_node); in QuantizePlaceholders()
361 SetNodeAttr("T", DT_QUINT8, &dequantize_node); in QuantizePlaceholders()
362 SetNodeAttr("mode", "MIN_FIRST", &dequantize_node); in QuantizePlaceholders()
420 SetNodeAttr("T", DT_QINT32, &quantize_node); in ConvertFakeQuantsToRequantize()
421 SetNodeAttr("mode", "MIN_FIRST", &quantize_node); in ConvertFakeQuantsToRequantize()
430 SetNodeAttr("Tinput", DT_QINT32, &requantize_node); in ConvertFakeQuantsToRequantize()
431 SetNodeAttr("out_type", DT_QUINT8, &requantize_node); in ConvertFakeQuantsToRequantize()
443 SetNodeAttr("T", DT_QUINT8, &dequantize_node); in ConvertFakeQuantsToRequantize()
[all …]
Dfold_old_batch_norms_test.cc97 SetNodeAttr("T", DT_FLOAT, &batch_norm_node); in TestFoldOldBatchNorms()
98 SetNodeAttr("variance_epsilon", 0.00001f, &batch_norm_node); in TestFoldOldBatchNorms()
99 SetNodeAttr("scale_after_normalization", false, &batch_norm_node); in TestFoldOldBatchNorms()
175 SetNodeAttr("T", DT_FLOAT, &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
176 SetNodeAttr("variance_epsilon", 0.00001f, &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
177 SetNodeAttr("scale_after_normalization", false, &batch_norm_node); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
253 SetNodeAttr("T", DT_FLOAT, &batch_norm_node); in TestFoldFusedBatchNorms()
254 SetNodeAttr("epsilon", 0.00001f, &batch_norm_node); in TestFoldFusedBatchNorms()
255 SetNodeAttr("is_training", false, &batch_norm_node); in TestFoldFusedBatchNorms()
330 SetNodeAttr("T", DT_FLOAT, &batch_norm_node); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative()
[all …]
Dbackports_test.cc81 SetNodeAttr("Tidx", DT_INT32, concat_node); in TestBackportConcatV2()
122 SetNodeAttr("dtype", DT_FLOAT, tensor_array_node); in TEST()
123 SetNodeAttr("element_shape", TensorShape({1, 2}), tensor_array_node); in TEST()
124 SetNodeAttr("dynamic_size", false, tensor_array_node); in TEST()
125 SetNodeAttr("clear_after_read", true, tensor_array_node); in TEST()
126 SetNodeAttr("tensor_array_name", "some_name", tensor_array_node); in TEST()
143 SetNodeAttr("source", "foo", tensor_array_grad_node); in TEST()
Dadd_default_attributes_test.cc48 SetNodeAttr("depth_radius", 7, lrn_node2); in TestAddDefaultAttributes()
49 SetNodeAttr("bias", 2.0f, lrn_node2); in TestAddDefaultAttributes()
50 SetNodeAttr("alpha", 2.0f, lrn_node2); in TestAddDefaultAttributes()
51 SetNodeAttr("beta", 1.0f, lrn_node2); in TestAddDefaultAttributes()
Dsparsify_gather.cc362 SetNodeAttr("dtype", key_dtype, &indices_node); in SparsifyGatherInternal()
367 SetNodeAttr("dtype", data_type, &values_node); in SparsifyGatherInternal()
373 SetNodeAttr("key_dtype", key_dtype, &hashtable_node); in SparsifyGatherInternal()
374 SetNodeAttr("value_dtype", data_type, &hashtable_node); in SparsifyGatherInternal()
381 SetNodeAttr("Tkey", key_dtype, &init_table_node); in SparsifyGatherInternal()
382 SetNodeAttr("Tval", data_type, &init_table_node); in SparsifyGatherInternal()
389 SetNodeAttr("Tin", key_dtype, &lookup_node); in SparsifyGatherInternal()
390 SetNodeAttr("Tout", data_type, &lookup_node); in SparsifyGatherInternal()
398 SetNodeAttr("dtype", data_type, &default_value_node); in SparsifyGatherInternal()
407 SetNodeAttr("value", dim_idx, &dim_idx_node); in SparsifyGatherInternal()
[all …]
Dquantize_weights.cc103 SetNodeAttr("dtype", DT_QUINT8, &quantized_const_node); in QuantizeWeights()
111 SetNodeAttr("dtype", DT_FLOAT, &min_node); in QuantizeWeights()
120 SetNodeAttr("dtype", DT_FLOAT, &max_node); in QuantizeWeights()
129 SetNodeAttr("T", DT_QUINT8, &dequantize_node); in QuantizeWeights()
130 SetNodeAttr("mode", "MIN_FIRST", &dequantize_node); in QuantizeWeights()
Dremove_attribute_test.cc46 SetNodeAttr<int32>("foo", 23, mul_node1); in TestRemoveAttribute()
47 SetNodeAttr<string>("bar", "something", mul_node1); in TestRemoveAttribute()
54 SetNodeAttr<int32>("foo", 46, add_node2); in TestRemoveAttribute()
55 SetNodeAttr<int32>("bob", 23, add_node2); in TestRemoveAttribute()
56 SetNodeAttr<string>("bar", "something else", add_node2); in TestRemoveAttribute()
Dinsert_logging.cc126 SetNodeAttr("message", node_message, print_node); in InsertLogging()
127 SetNodeAttr("first_n", first_n, print_node); in InsertLogging()
128 SetNodeAttr("summarize", summarize, print_node); in InsertLogging()
130 SetNodeAttr("T", output_types[0], print_node); in InsertLogging()
134 SetNodeAttr("U", output_types, print_node); in InsertLogging()
Dfuse_quantized_convolution.cc112 SetNodeAttr("Tsummand", DT_QINT8, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
114 SetNodeAttr("Tsummand", DT_QUINT8, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
128 SetNodeAttr("out_type", DT_QINT8, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
130 SetNodeAttr("out_type", DT_QUINT8, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
210 SetNodeAttr("Tbias", DT_QINT32, const_cast<NodeDef*>(node)); in FuseQuantizedConvolutionAndRequantize()
212 SetNodeAttr("Tbias", DT_FLOAT, const_cast<NodeDef*>(node)); in FuseQuantizedConvolutionAndRequantize()
Dflatten_atrous.cc101 SetNodeAttr("dtype", DT_FLOAT, &upsampled_filter_node); in FlattenAtrousConv()
117 SetNodeAttr("padding", "SAME", &flattened_conv_node); in FlattenAtrousConv()
Dsparsify_gather_test.cc133 SetNodeAttr("dtype", DT_FLOAT, w_node); in TestSinglePartition()
155 SetNodeAttr("N", 2, concat_node); in TestSinglePartition()
163 SetNodeAttr("N", 3, concat_node); in TestSinglePartition()
371 SetNodeAttr("dtype", DT_FLOAT, w_node1); in TestMultiPartition()
372 SetNodeAttr("dtype", DT_FLOAT, w_node2); in TestMultiPartition()
386 SetNodeAttr("N", 2, concat_node); in TestMultiPartition()
Dstrip_unused_nodes.cc181 SetNodeAttr("dtype", type, &placeholder_node); in StripUnusedNodes()
182 SetNodeAttr("shape", shape, &placeholder_node); in StripUnusedNodes()
Dfuse_convolutions.cc110 SetNodeAttr("dtype", DT_INT32, &pad_dims_node); in FuseResizeAndConv()
125 SetNodeAttr("mode", "REFLECT", &fused_conv); in FuseResizeAndConv()
Dfreeze_requantization_ranges.cc183 SetNodeAttr("dtype", DT_FLOAT, min_node); in FreezeRequantizationRanges()
193 SetNodeAttr("dtype", DT_FLOAT, max_node); in FreezeRequantizationRanges()
Dtransform_utils.h76 inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) { in SetNodeAttr() function
88 SetNodeAttr(key, tensor_proto, node); in SetNodeTensorAttr()
Dfold_batch_norms.cc105 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FoldBatchNorms()
Dround_weights.cc101 SetNodeAttr("dtype", DT_FLOAT, &rounded_const_node); in RoundWeights()
Dbackports.cc90 SetNodeAttr("dtype", DT_FLOAT, &replacement_flow_node); in BackportTensorArrayV3Transform()
Dfold_old_batch_norms.cc147 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
159 SetNodeAttr("dtype", DT_FLOAT, &bias_offset_node); in FuseScaleOffsetToConvWeights()
Dtransform_utils_test.cc176 SetNodeAttr("foo", value_i, &node); in TestSetNodeAttr()
179 SetNodeAttr("bar", value_s, &node); in TestSetNodeAttr()
214 SetNodeAttr("foo", tensor_proto, &node); in TestGetNodeTensorAttr()
DREADME.md984 SetNodeAttr("dtype", DT_INT32, &pad_dims_node);
999 SetNodeAttr("mode", "REFLECT", &fused_conv);
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc379 inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) { in SetNodeAttr() function
386 inline void SetNodeAttr(const string& key, const Tensor& tensor, in SetNodeAttr() function
390 SetNodeAttr(key, tensor_proto, node); in SetNodeAttr()
404 SetNodeAttr("key_dtype", key_dtype, hashtable_node); in TEST_F()
405 SetNodeAttr("value_dtype", data_dtype, hashtable_node); in TEST_F()
411 SetNodeAttr("dtype", key_dtype, keys_node); in TEST_F()
415 SetNodeAttr("value", keys, keys_node); in TEST_F()
420 SetNodeAttr("dtype", data_dtype, values_node); in TEST_F()
424 SetNodeAttr("value", values, values_node); in TEST_F()
430 SetNodeAttr("Tkey", key_dtype, init_table_node); in TEST_F()
[all …]