MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SessionOptions

Class SessionOptions

tensorflow/core/public/session_options.h:29–62  ·  view source on GitHub ↗

Configuration information for a Session.

Source from the content-addressed store, hash-verified

27const string CUDA_GRAPH_MODE_TARGET_NAME = "CudaGraphModeSession";
28/// Configuration information for a Session.
29struct SessionOptions {
30 /// The environment to use.
31 Env* env;
32
33 /// \brief The TensorFlow runtime to connect to.
34 ///
35 /// If 'target' is empty or unspecified, the local TensorFlow runtime
36 /// implementation will be used. Otherwise, the TensorFlow engine
37 /// defined by 'target' will be used to perform all computations.
38 ///
39 /// "target" can be either a single entry or a comma separated list
40 /// of entries. Each entry is a resolvable address of the
41 /// following format:
42 /// local
43 /// ip:port
44 /// host:port
45 /// ... other system-specific formats to identify tasks and jobs ...
46 ///
47 /// NOTE: at the moment 'local' maps to an in-process service-based
48 /// runtime.
49 ///
50 /// Upon creation, a single session affines itself to one of the
51 /// remote processes, with possible load balancing choices when the
52 /// "target" resolves to a list of possible processes.
53 ///
54 /// If the session disconnects from the remote process during its
55 /// lifetime, session calls may fail immediately.
56 string target;
57
58 /// Configuration options.
59 ConfigProto config;
60
61 SessionOptions();
62};
63
64struct CudaGraphModeSessionOptions: SessionOptions {
65 int batch_size = 0;

Callers 15

CandidateGraphRunnerMethod · 0.50
LoadGraphFunction · 0.50
LoadGraphFunction · 0.50
XrtClientTestMethod · 0.50
XlaCompilerMethod · 0.50
mainFunction · 0.50
SetUpMethod · 0.50
TESTFunction · 0.50
SetUpMethod · 0.50
RunAndFetchMethod · 0.50

Calls

no outgoing calls

Tested by 14

XrtClientTestMethod · 0.40
mainFunction · 0.40
SetUpMethod · 0.40
TESTFunction · 0.40
SetUpMethod · 0.40
RunAndFetchMethod · 0.40
VerifyFusedBatchNormMethod · 0.40
TEST_FFunction · 0.40
TEST_FFunction · 0.40
RemoteMgrTestMethod · 0.40