diff options
Diffstat (limited to 'tensorflow/core/public/session.h')
| -rw-r--r-- | tensorflow/core/public/session.h | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/tensorflow/core/public/session.h b/tensorflow/core/public/session.h index 6d364edef3..cbaaf602ee 100644 --- a/tensorflow/core/public/session.h +++ b/tensorflow/core/public/session.h @@ -114,6 +114,33 @@ class Session { const std::vector<string>& target_node_names, std::vector<Tensor>* outputs) = 0; + /// \brief Sets up a graph for partial execution. All future feeds and + /// fetches are specified by 'input_names' and 'output_names'. Returns + /// 'handle' that can be used to perform a sequence of partial feeds and + /// fetches. + /// NOTE: This API is still experimental and may change. + virtual Status PRunSetup(const std::vector<string>& input_names, + const std::vector<string>& output_names, + const std::vector<string>& target_nodes, + string* handle) { + return errors::Unimplemented( + "Partial run is not supported for" + " this session."); + } + + /// \brief Continues the pending execution specified by 'handle' with the + /// provided input tensors and fills `outputs` for the endpoints specified + /// in `output_names`. + /// NOTE: This API is still experimental and may change. + virtual Status PRun(const string& handle, + const std::vector<std::pair<string, Tensor> >& inputs, + const std::vector<string>& output_names, + std::vector<Tensor>* outputs) { + return errors::Unimplemented( + "Partial run is not supported for" + " this session."); + } + /// \brief Closes this session. /// /// Closing a session releases the resources used by this session |