Wrapper for TF_PRunSetup that converts the arguments to appropriate types. If *out_status is OK, the caller becomes the owner of *out_handle.
| 317 | // Wrapper for TF_PRunSetup that converts the arguments to appropriate types. |
| 318 | // If *out_status is OK, the caller becomes the owner of *out_handle. |
| 319 | void TF_PRunSetup_wrapper(TF_DeprecatedSession* session, |
| 320 | const NameVector& input_names, |
| 321 | const NameVector& output_names, |
| 322 | const NameVector& target_nodes, TF_Status* out_status, |
| 323 | const char** out_handle) { |
| 324 | Py_BEGIN_ALLOW_THREADS; |
| 325 | TF_PRunSetup( |
| 326 | session, const_cast<const char**>(input_names.data()), input_names.size(), |
| 327 | const_cast<const char**>(output_names.data()), output_names.size(), |
| 328 | const_cast<const char**>(target_nodes.data()), target_nodes.size(), |
| 329 | out_handle, out_status); |
| 330 | Py_END_ALLOW_THREADS; |
| 331 | } |
| 332 | |
| 333 | // Wrapper for TF_PRun that converts the arguments to appropriate types. |
| 334 | // If *out_status is OK, the caller becomes the owner of the PyObjects |
nothing calls this directly
no test coverage detected