Configuration information for a Session.
| 27 | const string CUDA_GRAPH_MODE_TARGET_NAME = "CudaGraphModeSession"; |
| 28 | /// Configuration information for a Session. |
| 29 | struct 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 | |
| 64 | struct CudaGraphModeSessionOptions: SessionOptions { |
| 65 | int batch_size = 0; |
no outgoing calls