/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},
|