| 389 | } |
| 390 | |
| 391 | static TFE_TensorHandle* createTFEDequeue(TFE_Context* ctx, |
| 392 | TF_DataType inputType, |
| 393 | TFE_TensorHandle* queue, |
| 394 | TF_Status* status) { |
| 395 | TFE_Op* op = TFE_NewOp(ctx, "QueueDequeueV2", status); |
| 396 | if (!status->status.ok()) return nullptr; |
| 397 | std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op_deleter(op, TFE_DeleteOp); |
| 398 | TFE_OpSetDevice(op, DEFAULT_CPU_DEVICE, status); |
| 399 | if (!status->status.ok()) return nullptr; |
| 400 | |
| 401 | TFE_OpAddInput(op, queue, status); |
| 402 | if (!status->status.ok()) return nullptr; |
| 403 | TFE_OpSetAttrTypeList(op, "component_types", &inputType, 1); |
| 404 | TFE_OpSetAttrInt(op, "timeout_ms", -1); |
| 405 | TFE_TensorHandle* ret; |
| 406 | int num_retvals = 1; |
| 407 | TFE_Execute(op, &ret, &num_retvals, status); |
| 408 | if (!status->status.ok()) return nullptr; |
| 409 | CHECK_EQ(num_retvals, 1); |
| 410 | return ret; |
| 411 | } |
| 412 | |
| 413 | TFE_TensorHandle* TFE_DequeueNamedTensor(TF_Session* session, int tensor_id, |
| 414 | TF_DataType inputType, |
no test coverage detected