experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #26229 from Intel-tensorflow:requantize_support_int8

Browse Source 
PiperOrigin-RevId: 239448231
This commit is contained in:
TensorFlower Gardener 2019-03-20 12:38:39 -07:00
commit d6de125302
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/kernels/requantize.cc
View File

@ -18,7 +18,6 @@ limitations under the License.
#define EIGEN_USE_THREADS
#include <math.h>
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/op_kernel.h"
@ -100,4 +99,10 @@ REGISTER_KERNEL_BUILDER(Name("Requantize")
.TypeConstraint<quint8>("out_type"),
RequantizeOp<qint32, quint8>);
REGISTER_KERNEL_BUILDER(Name("Requantize")
.Device(DEVICE_CPU)
.TypeConstraint<qint32>("Tinput")
.TypeConstraint<qint8>("out_type"),
RequantizeOp<qint32, qint8>);
} // namespace tensorflow