Searched refs:operand_states_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.h | 164 return states_[operand_states_[{op, index}]].params; in GetOperandParams()
190 return states_[operand_states_[{op, index}]]; in GetOperandQuantState()
200 return rescale_states_[operand_states_[{op, index}]]; in GetOperandRequantizeState()
225 llvm::DenseMap<OpValue, int> operand_states_; variable
|
| D | quantization_driver.cc | 213 return states_[operand_states_[{op, index}]]; in GetOperandQuantState()
227 return rescale_states_[operand_states_[{op, index}]]; in GetOperandRequantizeState()
271 operand_states_.insert({{op, index}, cached.first->second}); in InitializeOperandState()
378 llvm::DenseMap<OpValue, int> operand_states_; member in mlir::quant::__anonf70393e20111::QuantizationDriver
419 operand_states_.insert({{op, index}, next_state_index}); in InitializeState()
|
| D | quantization_context.cc | 296 operand_states_.insert({{op, index}, next_state_index}); in InitializeState()
306 operand_states_.insert({{op, index}, cached.first->second}); in InitializeOperandState()
|