Searched refs:quantization_ranges_ (Results 1 – 3 of 3) sorted by relevance
1295 quantization_ranges_[tensor] = symmetric_range; in ProvideQuantizationRange()1305 for (auto pair : quantization_ranges_) { in MaybeApplyQuantizationRanges()1334 if (!quantization_ranges_.count(tensor)) { in MaybeApplyQuantizationRanges()1359 auto input_tensor_range = quantization_ranges_.find(it->first); in PropagateQuantizationRanges()1360 auto output_tensor_range = quantization_ranges_.find(it->second); in PropagateQuantizationRanges()1361 if (input_tensor_range != quantization_ranges_.end() && in PropagateQuantizationRanges()1362 output_tensor_range == quantization_ranges_.end()) { in PropagateQuantizationRanges()1366 quantization_ranges_[it->second] = input_tensor_range->second; in PropagateQuantizationRanges()1372 if (quantization_ranges_.find(it->second) != quantization_ranges_.end()) { in PropagateQuantizationRanges()
561 std::unordered_map<nvinfer1::ITensor*, float> quantization_ranges_; variable
564 return converter_->quantization_ranges_; in quantization_ranges()1279 return converter_->quantization_ranges_; in quantization_ranges()