Searched refs:rescale_states_ (Results 1 – 2 of 2) sorted by relevance
200 return rescale_states_[operand_states_[{op, index}]]; in GetOperandRequantizeState()205 return rescale_states_[result_states_[{op, index}]]; in GetResultRequantizeState()221 std::unordered_map<int, RequantizeState> rescale_states_; variable
227 return rescale_states_[operand_states_[{op, index}]]; in GetOperandRequantizeState()232 return rescale_states_[result_states_[{op, index}]]; in GetResultRequantizeState()236 return rescale_states_[arg_states_[arg]]; in GetArgRequantizeState()374 std::unordered_map<int, RequantizeState> rescale_states_; member in mlir::quant::__anonf70393e20111::QuantizationDriver