diff options
| author |  Adrian Kuegel <akuegel@google.com> | 2018-09-19 02:28:02 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 02:35:28 -0700 |
| commit | edae2fe261b4395c6b84a8d24b48abd37b14f041 (patch) | |
| tree | 5fb61fa888f9a7f988d99778175ce10e1e15a8ac /tensorflow/compiler/tf2xla | |
| parent | 4732df127d13ce3af5840607b087fc79d883601b (diff) | |
Enable XlaSort and TopKV2 for CPU backend.
PiperOrigin-RevId: 213595499
Diffstat (limited to 'tensorflow/compiler/tf2xla')
| -rw-r--r-- | tensorflow/compiler/tf2xla/xla_cpu_backend.cc | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/tensorflow/compiler/tf2xla/xla_cpu_backend.cc b/tensorflow/compiler/tf2xla/xla_cpu_backend.cc index 23d04d43b3..ead229aacc 100644 --- a/tensorflow/compiler/tf2xla/xla_cpu_backend.cc +++ b/tensorflow/compiler/tf2xla/xla_cpu_backend.cc @@ -31,10 +31,6 @@ bool CpuOpFilter(KernelDef* kdef) { DT_FLOAT); return true; } - // TODO(b/26783907): The CPU backend currently does not implement sort. - if (kdef->op() == "XlaSort" || kdef->op() == "TopKV2") { - return false; - } if (kdef->op() == "Const") { AddDtypeToKernalDefConstraint("dtype", DT_STRING, kdef); } |