| 891 | } |
| 892 | |
| 893 | void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, |
| 894 | TF_Status* status) { |
| 895 | VLOG(1) << "Calling TFE_Execute() on op " << op; |
| 896 | tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> handle_retvals( |
| 897 | *num_retvals); |
| 898 | status->status = |
| 899 | tensorflow::EagerExecute(&op->operation, &handle_retvals, num_retvals); |
| 900 | if (!status->status.ok()) { |
| 901 | return; |
| 902 | } |
| 903 | for (int i = 0; i < *num_retvals; ++i) { |
| 904 | retvals[i] = new TFE_TensorHandle(handle_retvals[i]); |
| 905 | } |
| 906 | } |
| 907 | |
| 908 | TFE_TensorHandle* TFE_TensorHandleCopyToDevice(TFE_TensorHandle* h, |
| 909 | TFE_Context* ctx, |