Home
last modified time | relevance | path

Searched refs:GrpcSession (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc39 GrpcSession::GrpcSession(const SessionOptions& options) in GrpcSession() function in tensorflow::GrpcSession
42 GrpcSession::~GrpcSession() {} in ~GrpcSession()
45 Status GrpcSession::Create(const SessionOptions& options, in Create()
46 std::unique_ptr<GrpcSession>* out_session) { in Create()
47 std::unique_ptr<GrpcSession> session(new GrpcSession(options)); in Create()
96 void GrpcSession::SetHandleAndGraphVersion(string handle, int64 graph_version) { in SetHandleAndGraphVersion()
102 Status GrpcSession::Handle(string* out_handle) { in Handle()
111 Status GrpcSession::CreateImpl(CallOptions* call_options, in CreateImpl()
132 Status GrpcSession::Create(const GraphDef& graph) { in Create()
138 Status GrpcSession::Create(const RunOptions& run_options, in Create()
[all …]
Dgrpc_session.h49 class GrpcSession : public Session {
51 explicit GrpcSession(const SessionOptions& options);
55 std::unique_ptr<GrpcSession>* out_session);
60 ~GrpcSession() override;
140 TF_DISALLOW_COPY_AND_ASSIGN(GrpcSession);
Dgrpc_testlib.cc76 std::unique_ptr<GrpcSession> session; in MakeTestCluster()
77 TF_RETURN_IF_ERROR(GrpcSession::Create(options_copy, &session)); in MakeTestCluster()
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc97 std::unique_ptr<GrpcSession> session; in MakeGRPCCluster()
98 TF_CHECK_OK(GrpcSession::Create(options_copy, &session)); in MakeGRPCCluster()