diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-10 10:22:57 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 10:29:57 -0700 |
| commit | d2fa11acfd6b8a2a2663dc70cd899433bdde23e1 (patch) | |
| tree | 55d9865cef5c19fec51556aa1b8b5afc66702efa /tensorflow/contrib/lite/kernels/register.cc | |
| parent | a6df304e5ec4b0ad28884025558ecda414b9df07 (diff) | |
Adding FakeQuant op support.
PiperOrigin-RevId: 203969441
Diffstat (limited to 'tensorflow/contrib/lite/kernels/register.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/register.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/register.cc b/tensorflow/contrib/lite/kernels/register.cc index df88a66d19..1994e85ce3 100644 --- a/tensorflow/contrib/lite/kernels/register.cc +++ b/tensorflow/contrib/lite/kernels/register.cc @@ -103,6 +103,7 @@ TfLiteRegistration* Register_SQRT(); TfLiteRegistration* Register_RSQRT(); TfLiteRegistration* Register_SHAPE(); TfLiteRegistration* Register_POW(); +TfLiteRegistration* Register_FAKE_QUANT(); BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_RELU, Register_RELU()); @@ -189,6 +190,7 @@ BuiltinOpResolver::BuiltinOpResolver() { AddBuiltin(BuiltinOperator_RSQRT, Register_RSQRT()); AddBuiltin(BuiltinOperator_SHAPE, Register_SHAPE()); AddBuiltin(BuiltinOperator_POW, Register_POW()); + AddBuiltin(BuiltinOperator_FAKE_QUANT, Register_FAKE_QUANT()); // TODO(andrewharp, ahentz): Move these somewhere more appropriate so that // custom ops aren't always included by default. |