diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/tooling_util.h')
-rw-r--r-- | tensorflow/contrib/lite/toco/tooling_util.h | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/tooling_util.h b/tensorflow/contrib/lite/toco/tooling_util.h index d3b7224fe3..259ee7fbd0 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.h +++ b/tensorflow/contrib/lite/toco/tooling_util.h @@ -169,10 +169,23 @@ void GetQuantizationParamsFromMinMax(const MinMax& minmax, ::tflite::ChooseQuantizationParams<Integer>(rmin, rmax); } +template <typename T> +T ConvertOperator(Operator* o, OperatorType type) { + if (o != nullptr && o->type == type) { + return static_cast<T>(o); + } + + return nullptr; +} + void CheckIsReadyForQuantization(const Model& model); void UseDefaultMinMaxRangeValues(Model* model, double default_ranges_min, double default_ranges_max); +bool ReshapeIsEquivalentToTranspose(const Model& model, + const TensorFlowReshapeOperator* op, + bool allow_extra_unary_dims); + inline int Offset(const Shape& shape, const std::vector<int>& indices) { DCHECK_EQ(shape.dimensions_count(), indices.size()); const int dims_count = shape.dimensions_count(); |