Clarify use case of OperatorProperty::restrict_scale
PiperOrigin-RevId: 345022975 Change-Id: I528c3f0812e043528a64a27cced27eaa93a2fa13
This commit is contained in:
parent
71b41b8244
commit
2de460a15c
@ -110,7 +110,8 @@ struct OperatorProperty {
|
|||||||
|
|
||||||
// Use same min of min and max of max for each group.
|
// Use same min of min and max of max for each group.
|
||||||
// Incompatible with restrict_same_input_output_scale and restricted_value.
|
// Incompatible with restrict_same_input_output_scale and restricted_value.
|
||||||
// TODO(jianlijianli): make it compatible with other restrictions when there
|
// Currently it only supports scale pair of {input_index, output_index}.
|
||||||
|
// TODO(b/174534943): make it compatible with other restrictions when there
|
||||||
// is a use case.
|
// is a use case.
|
||||||
std::vector<std::vector<int>> restrict_scale = {};
|
std::vector<std::vector<int>> restrict_scale = {};
|
||||||
|
|
||||||
|
@ -890,7 +890,7 @@ TfLiteStatus QuantizeSharedRange(ModelT* model, ErrorReporter* error_reporter) {
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
// Currently only support pair of twos.
|
// Currently only support pair of twos.
|
||||||
// TODO(jianlijianli): extend to arbitrary number of tensors.
|
// TODO(b/174534943): extend to arbitrary number of tensors.
|
||||||
if (input.size() != 2) {
|
if (input.size() != 2) {
|
||||||
return kTfLiteError;
|
return kTfLiteError;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user