Searched refs:quant_conversion_map_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | delegate.cc | 184 if (quant_conversion_map_.empty()) return absl::OkStatus(); in GetRequiredTemporaries()
188 if (quant_conversion_map_.find(index) != quant_conversion_map_.end()) { in GetRequiredTemporaries()
193 if (quant_conversion_map_.find(index) != quant_conversion_map_.end()) { in GetRequiredTemporaries()
215 const bool is_dequant_required = !quant_conversion_map_.empty(); in Invoke()
218 DequantizeInputs(context, input_indices_, quant_conversion_map_)); in Invoke()
224 QuantizeOutputs(context, output_indices_, quant_conversion_map_)); in Invoke()
262 quant_conversion_map_.clear(); in InitializeGraph()
265 &quant_conversion_map_)); in InitializeGraph()
353 absl::flat_hash_map<int, int> quant_conversion_map_; member in tflite::gpu::__anonfe059bf90111::DelegateKernel
|
| D | metal_delegate.mm | 206 if (quant_conversion_map_.find(tensor_index) != quant_conversion_map_.end()) {
207 tensor_index = quant_conversion_map_[tensor_index];
210 quant_conversion_map_.erase(tensor_index);
245 if (quant_conversion_map_.empty()) return absl::OkStatus();
249 if (quant_conversion_map_.find(index) != quant_conversion_map_.end()) {
254 if (quant_conversion_map_.find(index) != quant_conversion_map_.end()) {
268 quant_conversion_map_.clear();
270 RETURN_IF_ERROR(BuildFinalModel(context, delegate_params, &graph, &quant_conversion_map_));
302 quant_conversion_map_.find(tensor_index) != quant_conversion_map_.end()) {
303 tensor_index = quant_conversion_map_[tensor_index];
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | object_reader.cc | 116 quant_conversion_map_, graph_, tensor_idx, in ReadValueByTensorIdx()
196 if (quant_conversion_map_ != nullptr && in AddUpdate()
197 quant_conversion_map_->find(update_tensor_idx) != in AddUpdate()
198 quant_conversion_map_->end()) { in AddUpdate()
202 tensor_to_value_->at(quant_conversion_map_->at(update_tensor_idx)) = in AddUpdate()
|
| D | object_reader.h | 49 quant_conversion_map_(quant_conversion_map) {} in graph_()
107 absl::flat_hash_map<int, int>* quant_conversion_map_; variable
|