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

remove duplicate registration for softmax bf16 op

Browse Source
This commit is contained in:
nammbash 2020-06-05 16:09:52 -07:00
parent a39ac9de5f
commit a894d7a936
1 changed files with 1 additions and 3 deletions

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

@ -58,9 +58,7 @@ namespace tensorflow {
REGISTER_KERNEL_BUILDER( \
Name("_FusedMatMul").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp); \
REGISTER_KERNEL_BUILDER( \
Name("BatchMatMulV2").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp); \
REGISTER_KERNEL_BUILDER( \
Name("Softmax").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp);
Name("BatchMatMulV2").Device(DEVICE_CPU).TypeConstraint<T>("T"), NoOp);
TF_CALL_bfloat16(REGISTER_CPU);
#undef REGISTER_CPU