| 444 | } |
| 445 | |
| 446 | void TFE_EnqueueNamedTensor(TF_Session* session, int tensor_id, |
| 447 | TFE_TensorHandle* tensor, TF_Status* status) { |
| 448 | assert(session); |
| 449 | VLOG(1) << "Enqueuing data tensor with id " << tensor_id; |
| 450 | |
| 451 | auto ctx = TFE_CreateContextFromSession(session, status); |
| 452 | if (!status->status.ok()) return; |
| 453 | std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( |
| 454 | ctx, TFE_DeleteContext); |
| 455 | |
| 456 | TF_DataType inputType = TFE_TensorHandleDataType(tensor); |
| 457 | TFE_TensorHandle* queue = createTFEQueue(ctx, inputType, tensor_id, status); |
| 458 | if (!status->status.ok()) return; |
| 459 | std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> |
| 460 | queue_deleter(queue, TFE_DeleteTensorHandle); |
| 461 | |
| 462 | createTFEEnqueue(ctx, inputType, queue, tensor, status); |
| 463 | } |
| 464 | |
| 465 | void TFE_EnqueueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, |
| 466 | TFE_TensorHandle* tensor, |
nothing calls this directly
no test coverage detected