| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_weights.cc | 104 SetNodeTensorAttr<float>("value", quantized_tensor, in QuantizeWeights()
114 SetNodeTensorAttr<float>("value", min_tensor, &min_node); in QuantizeWeights()
123 SetNodeTensorAttr<float>("value", max_tensor, &max_node); in QuantizeWeights()
|
| D | sparsify_gather_test.cc | 66 SetNodeTensorAttr<int32>("value", axis_t, axis_node); in MakeGather()
93 SetNodeTensorAttr<float>("value", weights, w_node); in TestSinglePartition()
113 SetNodeTensorAttr<string>("value", tensor_names_values, in TestSinglePartition()
120 SetNodeTensorAttr<string>("value", shapes_slices_val, in TestSinglePartition()
314 SetNodeTensorAttr<float>("value", weights, w_node1); in TestMultiPartition()
315 SetNodeTensorAttr<float>("value", weights, w_node2); in TestMultiPartition()
324 SetNodeTensorAttr<string>("value", tensor_names_values, in TestMultiPartition()
332 SetNodeTensorAttr<string>("value", shapes_slices_val, in TestMultiPartition()
|
| D | transform_utils.h | 84 inline void SetNodeTensorAttr(const string& key, const Tensor& tensor, in SetNodeTensorAttr() function
93 inline void SetNodeTensorAttr(const string& key, const TensorShape& shape, in SetNodeTensorAttr() function
100 SetNodeTensorAttr<T>(key, tensor, node); in SetNodeTensorAttr()
|
| D | quantize_nodes.cc | 345 SetNodeTensorAttr<float>("value", min_tensor, &min_node); in QuantizePlaceholders()
354 SetNodeTensorAttr<float>("value", max_tensor, &max_node); in QuantizePlaceholders()
749 SetNodeTensorAttr<int32>("value", reshape_dims_tensor, &reshape_dims); in QuantizeNodes()
759 SetNodeTensorAttr<int32>("value", reduction_dims_tensor, in QuantizeNodes()
854 SetNodeTensorAttr<float>("value", fallback_min_tensor, in QuantizeNodes()
865 SetNodeTensorAttr<float>("value", fallback_max_tensor, in QuantizeNodes()
|
| D | freeze_requantization_ranges.cc | 186 SetNodeTensorAttr<float>("value", min_tensor, min_node); in FreezeRequantizationRanges()
196 SetNodeTensorAttr<float>("value", max_tensor, max_node); in FreezeRequantizationRanges()
|
| D | fold_batch_norms.cc | 106 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FoldBatchNorms()
|
| D | round_weights.cc | 102 SetNodeTensorAttr<float>("value", rounded_tensor, &rounded_const_node); in RoundWeights()
|
| D | flatten_atrous.cc | 102 SetNodeTensorAttr<float>("value", upsampled_filter, in FlattenAtrousConv()
|
| D | backports.cc | 97 SetNodeTensorAttr<float>("value", replacement_flow_tensor, in BackportTensorArrayV3Transform()
|
| D | fold_old_batch_norms.cc | 148 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
160 SetNodeTensorAttr<float>("value", bias_offset, &bias_offset_node); in FuseScaleOffsetToConvWeights()
|
| D | fuse_convolutions.cc | 111 SetNodeTensorAttr<int32>("value", {4, 2}, {0, 0, 0, 0, 0, 0, 0, 0}, in FuseResizeAndConv()
|
| D | backports_test.cc | 116 SetNodeTensorAttr<float>("value", size_tensor, size_node); in TEST()
|
| D | sparsify_gather.cc | 86 SetNodeTensorAttr<float>("value", tensor, node_def); in CreateConstNode()
|
| D | transform_utils_test.cc | 185 SetNodeTensorAttr<int32>("foo", {3, 1}, {1, 2, 3}, &node); in TestSetNodeTensorAttr()
201 SetNodeTensorAttr<int32>("foo", input_tensor, &node); in TestSetNodeTensorAttrWithTensor()
|
| D | README.md | 985 SetNodeTensorAttr<int32>("value", {4, 2}, {0, 0, 0, 0, 0, 0, 0, 0},
|