Home
last modified time | relevance | path

Searched refs:ProvideQuantizationRange (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc500 params->converter->ProvideQuantizationRange(*tensor, value, value);
1765 converter->ProvideQuantizationRange(*tensor, min_range, max_range); in PrepareTensorForShape()
1777 void Converter::ProvideQuantizationRange(nvinfer1::ITensor* tensor, in ProvideQuantizationRange() function in tensorflow::tensorrt::convert::Converter
3877 params->converter->ProvideQuantizationRange(output_tensor, clip_value_min, in ConvertClipByValue()
3936 params->converter->ProvideQuantizationRange(output_tensor, 0.0f, 1.0f); in ConvertActivation()
3938 params->converter->ProvideQuantizationRange(output_tensor, -1.0f, 1.0f); in ConvertActivation()
3940 params->converter->ProvideQuantizationRange(output_tensor, -1.0f, 1.0f); in ConvertActivation()
3992 params->converter->ProvideQuantizationRange(inputs.at(0).tensor(), min_range, in ConvertQuantize()
4025 params->converter->ProvideQuantizationRange(output_tensor, 0.0f, 6.0f); in ConvertRelu6()
4044 params->converter->ProvideQuantizationRange(relu_layer->getOutput(0), 0.0f, in ConvertRelu6()
[all …]
Dconvert_nodes.h511 void ProvideQuantizationRange(nvinfer1::ITensor* tensor, float min_range,
Dconvert_nodes_test.cc1073 TEST_F(ConverterTest, ProvideQuantizationRange) { in TEST_F() argument
1076 converter_->ProvideQuantizationRange(&fake_tensor, 0.0f, 6.0f); in TEST_F()
1078 converter_->ProvideQuantizationRange(&fake_tensor, 1.0f, 6.0f); in TEST_F()
1080 converter_->ProvideQuantizationRange(&fake_tensor, -8.0f, 6.0f); in TEST_F()
1082 converter_->ProvideQuantizationRange(&fake_tensor, -8.123f, -6.123f); in TEST_F()
1085 converter_->ProvideQuantizationRange(&fake_tensor, -6.123f, 6.123f); in TEST_F()
1101 int8_converter->ProvideQuantizationRange(&input, -5.0f, 5.0f); in TEST_F()
1102 int8_converter->ProvideQuantizationRange(&not_infer, -100.0f, 100.0f); in TEST_F()
1126 converter_->ProvideQuantizationRange(&infer[4], -5.0f, 5.0f); in TEST_F()
7159 converter_->ProvideQuantizationRange(input.tensor(), -5.0f, 5.0f); in TEST_P()