Home
last modified time | relevance | path

Searched refs:SetNodeTensorAttr (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_weights.cc104 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()
Dsparsify_gather_test.cc66 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()
Dtransform_utils.h84 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()
Dquantize_nodes.cc345 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()
Dfreeze_requantization_ranges.cc186 SetNodeTensorAttr<float>("value", min_tensor, min_node); in FreezeRequantizationRanges()
196 SetNodeTensorAttr<float>("value", max_tensor, max_node); in FreezeRequantizationRanges()
Dfold_batch_norms.cc106 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FoldBatchNorms()
Dround_weights.cc102 SetNodeTensorAttr<float>("value", rounded_tensor, &rounded_const_node); in RoundWeights()
Dflatten_atrous.cc102 SetNodeTensorAttr<float>("value", upsampled_filter, in FlattenAtrousConv()
Dbackports.cc97 SetNodeTensorAttr<float>("value", replacement_flow_tensor, in BackportTensorArrayV3Transform()
Dfold_old_batch_norms.cc148 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
160 SetNodeTensorAttr<float>("value", bias_offset, &bias_offset_node); in FuseScaleOffsetToConvWeights()
Dfuse_convolutions.cc111 SetNodeTensorAttr<int32>("value", {4, 2}, {0, 0, 0, 0, 0, 0, 0, 0}, in FuseResizeAndConv()
Dbackports_test.cc116 SetNodeTensorAttr<float>("value", size_tensor, size_node); in TEST()
Dsparsify_gather.cc86 SetNodeTensorAttr<float>("value", tensor, node_def); in CreateConstNode()
Dtransform_utils_test.cc185 SetNodeTensorAttr<int32>("foo", {3, 1}, {1, 2, 3}, &node); in TestSetNodeTensorAttr()
201 SetNodeTensorAttr<int32>("foo", input_tensor, &node); in TestSetNodeTensorAttrWithTensor()
DREADME.md985 SetNodeTensorAttr<int32>("value", {4, 2}, {0, 0, 0, 0, 0, 0, 0, 0},