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

Removed duplicated registration of Less with bfloat16

Browse Source 
This fix tries to address the issue raised in 30476 where
Op Less was registered twice which triggered `Multiple OpKernel registrations` error
This fix removes the duplication.

This fix fies 30476.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2019-07-08 00:22:04 -07:00
parent dcb3eec82f
commit 46b6cde864
1 changed files with 1 additions and 2 deletions

3
tensorflow/core/kernels/cwise_op_less.cc
View File

@ -18,8 +18,7 @@ limitations under the License.
namespace tensorflow {
REGISTER5(BinaryOp, CPU, "Less", functor::less, float, Eigen::half, double,
bfloat16, int32);
REGISTER5(BinaryOp, CPU, "Less", functor::less, int64, uint8, int8, int16,
bfloat16);
REGISTER4(BinaryOp, CPU, "Less", functor::less, int64, uint8, int8, int16);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
REGISTER7(BinaryOp, GPU, "Less", functor::less, float, Eigen::half, double,