Searched refs:quantization_ranges_ (Results 1 – 3 of 3) sorted by relevance
655 std::unordered_map<nvinfer1::ITensor*, float> quantization_ranges_; variable
1780 quantization_ranges_[tensor] = symmetric_range; in ProvideQuantizationRange()1825 for (auto pair : quantization_ranges_) { in MaybeApplyQuantizationRanges()1945 if (!quantization_ranges_.count(tensor) && in MaybeApplyQuantizationRanges()1983 auto input_tensor_range = quantization_ranges_.find(it->first); in PropagateQuantizationRanges()1984 auto output_tensor_range = quantization_ranges_.find(it->second); in PropagateQuantizationRanges()1985 if (input_tensor_range != quantization_ranges_.end() && in PropagateQuantizationRanges()1986 output_tensor_range == quantization_ranges_.end()) { in PropagateQuantizationRanges()1990 quantization_ranges_[it->second] = input_tensor_range->second; in PropagateQuantizationRanges()1996 if (quantization_ranges_.find(it->second) != quantization_ranges_.end()) { in PropagateQuantizationRanges()
782 return converter_->quantization_ranges_; in quantization_ranges()1701 return converter_->quantization_ranges_; in quantization_ranges()