Home
last modified time | relevance | path

Searched refs:RequantizeState (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_driver.cc65 struct RequantizeState { struct
198 void RequantizeOpResult(Operation *op, int index, RequantizeState *state);
200 void RequantizeArg(BlockArgument arg, RequantizeState *state);
204 void RequantizeValue(Value value, RequantizeState *state, Location loc);
226 RequantizeState &GetOperandRequantizeState(Operation *op, int index) { in GetOperandRequantizeState()
231 RequantizeState &GetResultRequantizeState(Operation *op, int index) { in GetResultRequantizeState()
235 RequantizeState &GetArgRequantizeState(BlockArgument arg) { in GetArgRequantizeState()
307 if (requantize_state.pos != RequantizeState::NO_REQUANTIZE) { in DumpStates()
319 if (requantize_state.pos != RequantizeState::NO_REQUANTIZE) { in DumpStates()
334 if (requantize_state.pos != RequantizeState::NO_REQUANTIZE) { in DumpStates()
[all …]
Dquantization_context.h50 struct RequantizeState { struct
199 RequantizeState &GetOperandRequantizeState(Operation *op, int index) { in GetOperandRequantizeState()
204 RequantizeState &GetResultRequantizeState(Operation *op, int index) { in GetResultRequantizeState()
221 std::unordered_map<int, RequantizeState> rescale_states_;
Dquantization_context.cc130 if (state.IsEmpty() && requantize.pos == RequantizeState::NO_REQUANTIZE) { in Finalize()
132 } else if (requantize.pos == RequantizeState::ON_OUTPUT) { in Finalize()
145 if (state.IsEmpty() && requantize.pos == RequantizeState::NO_REQUANTIZE) { in Finalize()
147 } else if (requantize.pos == RequantizeState::ON_INPUT) { in Finalize()
338 rescale.pos = RequantizeState::ON_INPUT; in SetResultParams()
355 rescale.pos = RequantizeState::ON_OUTPUT; in SetOperandParams()