diff options
| author |  Kay Zhu <kayzhu@google.com> | 2018-02-23 12:24:19 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-23 12:28:06 -0800 |
| commit | ee333be5d16ae39029f9c58a989a84089ffadb5d (patch) | |
| tree | 420200453b3a67c9025491a006c6cbf8c8ed964c /tensorflow/compiler/jit/xla_interpreter_device.cc | |
| parent | f412e5c71781003d2408c1082220dee6b140f632 (diff) | |
[TF:XLA] Fix a bug where executor's device_ordinal should be passed to AllocateShapedBuffer. Also enable C64 type for interpreter device.
PiperOrigin-RevId: 186805709
Diffstat (limited to 'tensorflow/compiler/jit/xla_interpreter_device.cc')
| -rw-r--r-- | tensorflow/compiler/jit/xla_interpreter_device.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/jit/xla_interpreter_device.cc b/tensorflow/compiler/jit/xla_interpreter_device.cc index 2614deefd8..a329451b14 100644 --- a/tensorflow/compiler/jit/xla_interpreter_device.cc +++ b/tensorflow/compiler/jit/xla_interpreter_device.cc @@ -25,8 +25,8 @@ namespace tensorflow { const char* const DEVICE_XLA_INTERPRETER = "XLA_INTERPRETER"; const char* const DEVICE_INTERPRETER_XLA_JIT = "XLA_INTERPRETER_JIT"; -constexpr std::array<DataType, 5> kExecAllTypes = { - {DT_INT32, DT_FLOAT, DT_BOOL, DT_DOUBLE, DT_INT64}}; +constexpr std::array<DataType, 6> kExecAllTypes = { + {DT_INT32, DT_INT64, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BOOL}}; class XlaInterpreterDeviceFactory : public DeviceFactory { public: |