/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | quantize_nodes.cc | 336 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 …]
|
D | fold_old_batch_norms_test.cc | 97 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 …]
|
D | backports_test.cc | 81 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()
|
D | add_default_attributes_test.cc | 48 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()
|
D | sparsify_gather.cc | 362 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 …]
|
D | quantize_weights.cc | 103 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()
|
D | remove_attribute_test.cc | 46 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()
|
D | insert_logging.cc | 126 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()
|
D | fuse_quantized_convolution.cc | 112 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()
|
D | flatten_atrous.cc | 101 SetNodeAttr("dtype", DT_FLOAT, &upsampled_filter_node); in FlattenAtrousConv() 117 SetNodeAttr("padding", "SAME", &flattened_conv_node); in FlattenAtrousConv()
|
D | sparsify_gather_test.cc | 133 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()
|
D | strip_unused_nodes.cc | 181 SetNodeAttr("dtype", type, &placeholder_node); in StripUnusedNodes() 182 SetNodeAttr("shape", shape, &placeholder_node); in StripUnusedNodes()
|
D | fuse_convolutions.cc | 110 SetNodeAttr("dtype", DT_INT32, &pad_dims_node); in FuseResizeAndConv() 125 SetNodeAttr("mode", "REFLECT", &fused_conv); in FuseResizeAndConv()
|
D | freeze_requantization_ranges.cc | 183 SetNodeAttr("dtype", DT_FLOAT, min_node); in FreezeRequantizationRanges() 193 SetNodeAttr("dtype", DT_FLOAT, max_node); in FreezeRequantizationRanges()
|
D | transform_utils.h | 76 inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) { in SetNodeAttr() function 88 SetNodeAttr(key, tensor_proto, node); in SetNodeTensorAttr()
|
D | fold_batch_norms.cc | 105 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FoldBatchNorms()
|
D | round_weights.cc | 101 SetNodeAttr("dtype", DT_FLOAT, &rounded_const_node); in RoundWeights()
|
D | backports.cc | 90 SetNodeAttr("dtype", DT_FLOAT, &replacement_flow_node); in BackportTensorArrayV3Transform()
|
D | fold_old_batch_norms.cc | 147 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FuseScaleOffsetToConvWeights() 159 SetNodeAttr("dtype", DT_FLOAT, &bias_offset_node); in FuseScaleOffsetToConvWeights()
|
D | transform_utils_test.cc | 176 SetNodeAttr("foo", value_i, &node); in TestSetNodeAttr() 179 SetNodeAttr("bar", value_s, &node); in TestSetNodeAttr() 214 SetNodeAttr("foo", tensor_proto, &node); in TestGetNodeTensorAttr()
|
D | README.md | 984 SetNodeAttr("dtype", DT_INT32, &pad_dims_node); 999 SetNodeAttr("mode", "REFLECT", &fused_conv);
|
/external/tensorflow/tensorflow/core/grappler/clusters/ |
D | single_machine_test.cc | 379 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 …]
|