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

Add a bfloat16 registration for MulNoNan

Browse Source 
PiperOrigin-RevId: 309434179
Change-Id: Ie9f17a9ef52171eede511710cd39d053a4ddc414
This commit is contained in:
Allen Lavoie 2020-05-01 10:47:05 -07:00 committed by TensorFlower Gardener
parent f19618050b
commit 81be829592
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/ops/math_ops.cc
View File

@ -464,7 +464,7 @@ REGISTER_OP("MulNoNan")
.Input("x: T") .Input("x: T")
.Input("y: T") .Input("y: T")
.Output("z: T") .Output("z: T")
.Attr("T: {half, float, double, complex64, complex128}") .Attr("T: {bfloat16, half, float, double, complex64, complex128}")
.SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn); .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn);
// Note: This op is not commutative w.r.t. to all its inputs. // Note: This op is not commutative w.r.t. to all its inputs.