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

Removed duplicated registration of LessEqual with bfloat16

Browse Source 
This is a follow up PR to 30479 where the duplicated registration of
Less with bfloat16 has been removed, but not LessEqual

This fix fixed duplicate bfloat16 entry of LessEqual as well.

This fix is related to 30476.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2019-08-08 14:11:19 -07:00
parent efd304f40e
commit 39c0c42b6a
1 changed files with 2 additions and 2 deletions

4
tensorflow/core/kernels/cwise_op_less_equal.cc
View File

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