| 494 | } |
| 495 | |
| 496 | TFE_TensorHandle* TFE_DequeueVariantTensor(TF_Session* session, int tensor_id, |
| 497 | TF_Status* status) { |
| 498 | VLOG(1) << "Dequeuing variant tensor with id " << tensor_id; |
| 499 | |
| 500 | auto ctx = TFE_CreateContextFromSession(session, status); |
| 501 | if (!status->status.ok()) return nullptr; |
| 502 | std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( |
| 503 | ctx, TFE_DeleteContext); |
| 504 | |
| 505 | TFE_TensorHandle* queue = createTFEQueue(ctx, TF_VARIANT, tensor_id, status); |
| 506 | if (!status->status.ok()) return nullptr; |
| 507 | std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> |
| 508 | queue_deleter(queue, TFE_DeleteTensorHandle); |
| 509 | |
| 510 | return createTFEDequeue(ctx, TF_VARIANT, queue, status); |
| 511 | } |
| 512 | |
| 513 | static void CheckOk(TF_Status* status) { |
| 514 | CHECK_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); |
nothing calls this directly
no test coverage detected