Home
last modified time | relevance | path

Searched refs:SetNodeAttr (Results 1 – 23 of 23) 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()
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.cc125 SetNodeAttr("message", node_message, print_node); in InsertLogging()
126 SetNodeAttr("first_n", first_n, print_node); in InsertLogging()
127 SetNodeAttr("summarize", summarize, print_node); in InsertLogging()
129 SetNodeAttr("T", output_types[0], print_node); in InsertLogging()
133 SetNodeAttr("U", output_types, print_node); in InsertLogging()
Dsparsify_gather.cc361 SetNodeAttr("dtype", key_dtype, &indices_node); in SparsifyGatherInternal()
366 SetNodeAttr("dtype", data_type, &values_node); in SparsifyGatherInternal()
372 SetNodeAttr("key_dtype", key_dtype, &hashtable_node); in SparsifyGatherInternal()
373 SetNodeAttr("value_dtype", data_type, &hashtable_node); in SparsifyGatherInternal()
380 SetNodeAttr("Tkey", key_dtype, &init_table_node); in SparsifyGatherInternal()
381 SetNodeAttr("Tval", data_type, &init_table_node); in SparsifyGatherInternal()
388 SetNodeAttr("Tin", key_dtype, &lookup_node); in SparsifyGatherInternal()
389 SetNodeAttr("Tout", data_type, &lookup_node); in SparsifyGatherInternal()
397 SetNodeAttr("dtype", data_type, &default_value_node); in SparsifyGatherInternal()
406 SetNodeAttr("value", dim_idx, &dim_idx_node); in SparsifyGatherInternal()
[all …]
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()
Dfuse_convolutions.cc109 SetNodeAttr("dtype", DT_INT32, &pad_dims_node); in FuseResizeAndConv()
124 SetNodeAttr("mode", "REFLECT", &fused_conv); in FuseResizeAndConv()
Dstrip_unused_nodes.cc180 SetNodeAttr("dtype", type, &placeholder_node); in StripUnusedNodes()
181 SetNodeAttr("shape", shape, &placeholder_node); in StripUnusedNodes()
Dfreeze_requantization_ranges.cc182 SetNodeAttr("dtype", DT_FLOAT, min_node); in FreezeRequantizationRanges()
192 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()
Dround_weights.cc101 SetNodeAttr("dtype", DT_FLOAT, &rounded_const_node); in RoundWeights()
Dfold_batch_norms.cc104 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FoldBatchNorms()
Dbackports.cc89 SetNodeAttr("dtype", DT_FLOAT, &replacement_flow_node); in BackportTensorArrayV3Transform()
Dfold_old_batch_norms.cc146 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
158 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/optimizers/inference/
Dbatch_op_rewriter_test.cc48 ::tensorflow::graph_transforms::SetNodeAttr("num_batch_threads", in AddBatchOp()
50 ::tensorflow::graph_transforms::SetNodeAttr("max_batch_size", 16, batch_op); in AddBatchOp()
51 ::tensorflow::graph_transforms::SetNodeAttr("batch_timeout_micros", 10000, in AddBatchOp()
53 ::tensorflow::graph_transforms::SetNodeAttr( in AddBatchOp()
55 ::tensorflow::graph_transforms::SetNodeAttr("max_enqueued_batches", 1000, in AddBatchOp()
Dbatch_op_rewriter.cc93 ::tensorflow::graph_transforms::SetNodeAttr(kNumBatchThreadsAttr, 0, in Optimize()
104 ::tensorflow::graph_transforms::SetNodeAttr(kNumBatchThreadsAttr, 0, in Optimize()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc393 inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) { in SetNodeAttr() function
400 inline void SetNodeAttr(const string& key, const Tensor& tensor, in SetNodeAttr() function
404 SetNodeAttr(key, tensor_proto, node); in SetNodeAttr()
418 SetNodeAttr("key_dtype", key_dtype, hashtable_node); in TEST_F()
419 SetNodeAttr("value_dtype", data_dtype, hashtable_node); in TEST_F()
425 SetNodeAttr("dtype", key_dtype, keys_node); in TEST_F()
429 SetNodeAttr("value", keys, keys_node); in TEST_F()
434 SetNodeAttr("dtype", data_dtype, values_node); in TEST_F()
438 SetNodeAttr("value", values, values_node); in TEST_F()
444 SetNodeAttr("Tkey", key_dtype, init_table_node); in TEST_F()
[all …]