tensorflow::serving::test_util::MockSession Class Reference

Inheritance diagram for tensorflow::serving::test_util::MockSession:

Collaboration diagram for tensorflow::serving::test_util::MockSession:

Public Member Functions
	MOCK_METHOD (::tensorflow::Status, Create,(const GraphDef &graph),(override))
	MOCK_METHOD (::tensorflow::Status, Extend,(const GraphDef &graph),(override))
	MOCK_METHOD (::tensorflow::Status, Run,((const std::vector< std::pair< string, Tensor >> &inputs), const std::vector< string > &output_names, const std::vector< string > &target_nodes, std::vector< Tensor > *outputs),(override))
	MOCK_METHOD (::tensorflow::Status, Run,(const RunOptions &run_options,(const std::vector< std::pair< string, Tensor >> &inputs), const std::vector< string > &output_names, const std::vector< string > &target_nodes, std::vector< Tensor > *outputs, RunMetadata *run_metadata),(override))
	MOCK_METHOD (::tensorflow::Status, Run,(const RunOptions &run_options,(const std::vector< std::pair< string, Tensor >> &inputs), const std::vector< string > &output_names, const std::vector< string > &target_nodes, std::vector< Tensor > *outputs, RunMetadata *run_metadata, const tensorflow::thread::ThreadPoolOptions &thread_pool_options),(override))
	MOCK_METHOD (::tensorflow::Status, PRunSetup,(const std::vector< string > &input_names, const std::vector< string > &output_names, const std::vector< string > &target_nodes, string *handle),(override))
	MOCK_METHOD (::tensorflow::Status, PRun,(const string &handle,(const std::vector< std::pair< string, Tensor >> &inputs), const std::vector< string > &output_names, std::vector< Tensor > *outputs),(override))
	MOCK_METHOD (::tensorflow::Status, ListDevices,(std::vector<::tensorflow::DeviceAttributes > *response),(override))
	MOCK_METHOD (::tensorflow::Status, Close,(),(override))

Detailed Description

Definition at line 28 of file mock_session.h.

---

The documentation for this class was generated from the following file:

• tensorflow_serving/core/test_util/mock_session.h