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.cc1104 MarkQuantizationRangesAsInferrable(tensor, layer->getOutput(0)); in RenameAndMarkOutputTensors()
1170 MarkQuantizationRangesAsInferrable(input_tensor, layer->getOutput(0)); in TransposeTensor()
1257 MarkQuantizationRangesAsInferrable( in PrepareTensorForShape()
1286 void Converter::MarkQuantizationRangesAsInferrable(nvinfer1::ITensor* input, in MarkQuantizationRangesAsInferrable() function in tensorflow::tensorrt::convert::Converter
1688 params->converter->MarkQuantizationRangesAsInferrable( in BinaryTensorOpWeight()
1890 params->converter->MarkQuantizationRangesAsInferrable( in ConvertConv2DHelper()
2452 params->converter->MarkQuantizationRangesAsInferrable(tensor, in ConvertStridedSliceHelper()
2701 params->converter->MarkQuantizationRangesAsInferrable( in ConvertPool()
2713 params->converter->MarkQuantizationRangesAsInferrable( in ConvertPool()
2770 params->converter->MarkQuantizationRangesAsInferrable( in ConvertLeakyRelu()
[all …]
Dconvert_nodes_test.cc847 int8_converter.MarkQuantizationRangesAsInferrable(&input, &infer_1); in TEST_F()
848 int8_converter.MarkQuantizationRangesAsInferrable(&infer_1, &infer_2); in TEST_F()
849 int8_converter.MarkQuantizationRangesAsInferrable(&infer_2, &infer_3); in TEST_F()
869 converter_->MarkQuantizationRangesAsInferrable(&infer[0], &infer[1]); in TEST_F()
870 converter_->MarkQuantizationRangesAsInferrable(&infer[1], &infer[2]); in TEST_F()
871 converter_->MarkQuantizationRangesAsInferrable(&infer[3], &infer[2]); in TEST_F()
872 converter_->MarkQuantizationRangesAsInferrable(&infer[4], &infer[1]); in TEST_F()
873 converter_->MarkQuantizationRangesAsInferrable(&infer[4], &infer[5]); in TEST_F()
Dconvert_nodes.h472 void MarkQuantizationRangesAsInferrable(nvinfer1::ITensor* input,