diff options
Diffstat (limited to 'tensorflow/core/kernels/quantized_reshape_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/quantized_reshape_op.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/quantized_reshape_op.cc b/tensorflow/core/kernels/quantized_reshape_op.cc index d49edd3feb..bd76c94ede 100644 --- a/tensorflow/core/kernels/quantized_reshape_op.cc +++ b/tensorflow/core/kernels/quantized_reshape_op.cc @@ -50,8 +50,8 @@ class QuantizedReshapeOp : public ReshapeOp { .TypeConstraint<type>("T"), \ QuantizedReshapeOp) -TF_CALL_quint8(REGISTER_CPU_KERNEL); -TF_CALL_qint32(REGISTER_CPU_KERNEL); +REGISTER_CPU_KERNEL(::tensorflow::quint8); +REGISTER_CPU_KERNEL(::tensorflow::qint32); #undef REGISTER_CPU_KERNEL |