Searched refs:max_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_weights.cc | 121 Tensor max_tensor(DT_FLOAT, {}); in QuantizeWeights() local
122 max_tensor.scalar<float>()() = max; in QuantizeWeights()
123 SetNodeTensorAttr<float>("value", max_tensor, &max_node); in QuantizeWeights()
|
| D | freeze_requantization_ranges.cc | 194 Tensor max_tensor(DT_FLOAT, {}); in FreezeRequantizationRanges() local
195 max_tensor.flat<float>()(0) = max_value; in FreezeRequantizationRanges()
196 SetNodeTensorAttr<float>("value", max_tensor, max_node); in FreezeRequantizationRanges()
|
| D | quantize_nodes.cc | 352 Tensor max_tensor(DT_FLOAT, {}); in QuantizePlaceholders() local
353 max_tensor.flat<float>()(0) = input_max; in QuantizePlaceholders()
354 SetNodeTensorAttr<float>("value", max_tensor, &max_node); in QuantizePlaceholders()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_add_op.cc | 292 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
299 tensor_data[i], min_tensor, max_tensor, output_min, output_max); in VectorTensorAddition()
310 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
323 QuantizedToFloat<quint8>(0, min_tensor, max_tensor); in VectorTensorAddition()
325 QuantizedToFloat<quint8>(1, min_tensor, max_tensor); in VectorTensorAddition()
395 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
408 QuantizedToFloat<quint8>(0, min_tensor, max_tensor); in VectorTensorAddition()
410 QuantizedToFloat<quint8>(1, min_tensor, max_tensor); in VectorTensorAddition()
|