| 411 | } |
| 412 | |
| 413 | TFE_TensorHandle* TFE_DequeueNamedTensor(TF_Session* session, int tensor_id, |
| 414 | TF_DataType inputType, |
| 415 | TF_Status* status) { |
| 416 | assert(session); |
| 417 | VLOG(1) << "Dequeuing data tensor with id " << tensor_id; |
| 418 | |
| 419 | auto ctx = TFE_CreateContextFromSession(session, status); |
| 420 | if (!status->status.ok()) return nullptr; |
| 421 | std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( |
| 422 | ctx, TFE_DeleteContext); |
| 423 | |
| 424 | TFE_TensorHandle* queue = createTFEQueue(ctx, inputType, tensor_id, status); |
| 425 | if (!status->status.ok()) return nullptr; |
| 426 | std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> |
| 427 | queue_deleter(queue, TFE_DeleteTensorHandle); |
| 428 | |
| 429 | auto* ret = createTFEDequeue(ctx, inputType, queue, status); |
| 430 | return ret; |
| 431 | } |
| 432 | |
| 433 | TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, |
| 434 | TF_DataType inputType, |
nothing calls this directly
no test coverage detected