| 631 | } |
| 632 | |
| 633 | const char* TFE_OpGetDevice(TFE_Op* op, TF_Status* status) { |
| 634 | tensorflow::Device* device = (op->operation.Device() == nullptr) |
| 635 | ? op->operation.EagerContext()->HostCPU() |
| 636 | : op->operation.Device(); |
| 637 | return device->name().c_str(); |
| 638 | } |
| 639 | |
| 640 | void TFE_OpSetXLACompilation(TFE_Op* op, unsigned char enable) { |
| 641 | op->operation.SetUseXla(enable); |