| 477 | } |
| 478 | |
| 479 | void TFE_EnqueueVariantTensor(TF_Session* session, int tensor_id, |
| 480 | TFE_TensorHandle* tensor, TF_Status* status) { |
| 481 | VLOG(1) << "Enqueuing variant tensor with id " << tensor_id; |
| 482 | |
| 483 | auto ctx = TFE_CreateContextFromSession(session, status); |
| 484 | if (!status->status.ok()) return; |
| 485 | std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( |
| 486 | ctx, TFE_DeleteContext); |
| 487 | |
| 488 | TFE_TensorHandle* queue = createTFEQueue(ctx, TF_VARIANT, tensor_id, status); |
| 489 | if (!status->status.ok()) return; |
| 490 | std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> |
| 491 | queue_deleter(queue, TFE_DeleteTensorHandle); |
| 492 | |
| 493 | createTFEEnqueue(ctx, TF_VARIANT, queue, tensor, status); |
| 494 | } |
| 495 | |
| 496 | TFE_TensorHandle* TFE_DequeueVariantTensor(TF_Session* session, int tensor_id, |
| 497 | TF_Status* status) { |
nothing calls this directly
no test coverage detected