Searched refs:SetResultParams (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.h | 89 bool SetResultParams(Operation *op, int index, QuantParams params) { in SetResultParams() function
90 return states_manager_.SetResultParams(op, index, params); in SetResultParams()
147 bool SetResultParams(Operation *op, int index, QuantParams params);
|
| D | quantization_driver.cc | 176 bool SetResultParams(Operation *op, int index, QuantParams params);
455 return SetResultParams(op, 0, quant_type); in SetConstantResultParams()
460 bool QuantizationDriver::SetResultParams(Operation *op, int res_index, in SetResultParams() function in mlir::quant::QuantizationDriver
843 changed |= SetResultParams(op, res, params); in PropagateParams()
857 changed |= SetResultParams(op, i, params); in PropagateParams()
|
| D | quantization_context.cc | 271 if (ele.isa<FloatType>() && SetResultParams(op, res, params)) { in PropagateQuantParams()
328 bool QuantizeContext::StatesManager::SetResultParams(Operation *op, in SetResultParams() function in mlir::quant::QuantizeContext::StatesManager
|