Searched refs:result_states_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.h | 159 return states_[result_states_[{op, index}]].params; in GetResultParams()
195 return states_[result_states_[{op, index}]]; in GetResultQuantState()
205 return rescale_states_[result_states_[{op, index}]]; in GetResultRequantizeState()
226 llvm::DenseMap<OpValue, int> result_states_; variable
|
| D | quantization_driver.cc | 218 return states_[result_states_[{op, index}]]; in GetResultQuantState()
232 return rescale_states_[result_states_[{op, index}]]; in GetResultRequantizeState()
284 result_states_.insert({{op, index}, cached.first->second}); in InitializeResultState()
379 llvm::DenseMap<OpValue, int> result_states_; member in mlir::quant::__anonf70393e20111::QuantizationDriver
417 result_states_.insert({{op, index}, next_state_index}); in InitializeState()
898 for (auto it : result_states_) { in Finalize()
|
| D | quantization_context.cc | 294 result_states_.insert({{op, index}, next_state_index}); in InitializeState()
317 result_states_.insert({{op, index}, cached.first->second}); in InitializeResultState()
|