Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1391 MarkQuantizationRangesAsInferrable(tensor, layer->getOutput(0)); in RenameAndMarkOutputTensors()
1601 MarkQuantizationRangesAsInferrable(input_tensor, layer->getOutput(0)); in TransposeTensor()
1742 converter->MarkQuantizationRangesAsInferrable(input.tensor(), in PrepareTensorForShape()
1771 void Converter::MarkQuantizationRangesAsInferrable(nvinfer1::ITensor* input, in MarkQuantizationRangesAsInferrable() function in tensorflow::tensorrt::convert::Converter
2244 params->converter->MarkQuantizationRangesAsInferrable( in Conv2DPaddingHelper()
3012 params->converter->MarkQuantizationRangesAsInferrable(tensor, in ConvertStridedSliceHelper()
3483 params->converter->MarkQuantizationRangesAsInferrable(tensor, in ConvertPool3D()
3743 params->converter->MarkQuantizationRangesAsInferrable( in ConvertPool()
3756 params->converter->MarkQuantizationRangesAsInferrable(tensor, in ConvertPool()
3832 params->converter->MarkQuantizationRangesAsInferrable( in ConvertLeakyRelu()
[all …]
Dconvert_nodes.h505 void MarkQuantizationRangesAsInferrable(nvinfer1::ITensor* input,
Dconvert_nodes_test.cc1103 int8_converter->MarkQuantizationRangesAsInferrable(&input, &infer_1); in TEST_F()
1104 int8_converter->MarkQuantizationRangesAsInferrable(&infer_1, &infer_2); in TEST_F()
1105 int8_converter->MarkQuantizationRangesAsInferrable(&infer_2, &infer_3); in TEST_F()
1127 converter_->MarkQuantizationRangesAsInferrable(&infer[0], &infer[1]); in TEST_F()
1128 converter_->MarkQuantizationRangesAsInferrable(&infer[1], &infer[2]); in TEST_F()
1129 converter_->MarkQuantizationRangesAsInferrable(&infer[3], &infer[2]); in TEST_F()
1130 converter_->MarkQuantizationRangesAsInferrable(&infer[4], &infer[1]); in TEST_F()
1131 converter_->MarkQuantizationRangesAsInferrable(&infer[4], &infer[5]); in TEST_F()