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.cc495 params->converter->ProvideQuantizationRange( in CreateBroadcastableScalarConstant()
1279 ProvideQuantizationRange(const_cast<nvinfer1::ITensor*>(*tensor), in PrepareTensorForShape()
1292 void Converter::ProvideQuantizationRange(nvinfer1::ITensor* tensor, in ProvideQuantizationRange() function in tensorflow::tensorrt::convert::Converter
2804 params->converter->ProvideQuantizationRange(output_tensor, 0.0f, 1.0f); in ConvertActivation()
2806 params->converter->ProvideQuantizationRange(output_tensor, -1.0f, 1.0f); in ConvertActivation()
2858 params->converter->ProvideQuantizationRange( in ConvertQuantize()
2901 params->converter->ProvideQuantizationRange(relu_layer->getOutput(0), 0.0f, in ConvertRelu6()
2919 params->converter->ProvideQuantizationRange(output_tensor, 0.0f, 6.0f); in ConvertRelu6()
3283 params->converter->ProvideQuantizationRange(output_tensor, -1.0f, 1.0f); in ConvertUnary()
3285 params->converter->ProvideQuantizationRange(output_tensor, -M_PI_2, M_PI_2); in ConvertUnary()
[all …]
Dconvert_nodes_test.cc823 TEST_F(ConverterTest, ProvideQuantizationRange) { in TEST_F() argument
826 converter_->ProvideQuantizationRange(&fake_tensor, 0.0f, 6.0f); in TEST_F()
828 converter_->ProvideQuantizationRange(&fake_tensor, 1.0f, 6.0f); in TEST_F()
830 converter_->ProvideQuantizationRange(&fake_tensor, -8.0f, 6.0f); in TEST_F()
832 converter_->ProvideQuantizationRange(&fake_tensor, -8.123f, -6.123f); in TEST_F()
835 converter_->ProvideQuantizationRange(&fake_tensor, -6.123f, 6.123f); in TEST_F()
845 int8_converter.ProvideQuantizationRange(&input, -5.0f, 5.0f); in TEST_F()
846 int8_converter.ProvideQuantizationRange(&not_infer, -100.0f, 100.0f); in TEST_F()
868 converter_->ProvideQuantizationRange(&infer[4], -5.0f, 5.0f); in TEST_F()
Dconvert_nodes.h478 void ProvideQuantizationRange(nvinfer1::ITensor* tensor, float min_range,