diff options
author | Benjamin Kramer <kramerb@google.com> | 2018-09-19 10:20:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 10:27:57 -0700 |
commit | 414ca1cda5aec72b48d5da127f61b0d05fbdc22c (patch) | |
tree | adf0ee8fe6b788c91609022accf92dfb432bb95d /tensorflow/compiler/tf2xla | |
parent | 0800a645b85fc9d7c18efe45d1006cf35fba93dd (diff) |
[XLA:CPU] Add an emitter for erfinv(double) and erfinv(half).
This is used by the random number generator. Same algorithm as for float, just with more
precision. fp16 is upcasted to fp32 and then processed with the float algorithm.
PiperOrigin-RevId: 213648736
Diffstat (limited to 'tensorflow/compiler/tf2xla')
-rw-r--r-- | tensorflow/compiler/tf2xla/xla_cpu_backend.cc | 11 |
1 files changed, 0 insertions, 11 deletions
diff --git a/tensorflow/compiler/tf2xla/xla_cpu_backend.cc b/tensorflow/compiler/tf2xla/xla_cpu_backend.cc index ead229aacc..bc44301d40 100644 --- a/tensorflow/compiler/tf2xla/xla_cpu_backend.cc +++ b/tensorflow/compiler/tf2xla/xla_cpu_backend.cc @@ -20,17 +20,6 @@ limitations under the License. namespace tensorflow { bool CpuOpFilter(KernelDef* kdef) { - // TODO(b/34339814): implement inverse erf for double types and remove this - // workaround. - if (kdef->op() == "RandomStandardNormal") { - kdef->clear_constraint(); - // Change the type constraint to permit only DTD_FLOAT. - KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); - attr_constraint->set_name("dtype"); - attr_constraint->mutable_allowed_values()->mutable_list()->add_type( - DT_FLOAT); - return true; - } if (kdef->op() == "Const") { AddDtypeToKernalDefConstraint("dtype", DT_STRING, kdef); } |