diff options
author | Manjunath Kudlur <keveman@gmail.com> | 2015-11-06 16:27:58 -0800 |
---|---|---|
committer | Manjunath Kudlur <keveman@gmail.com> | 2015-11-06 16:27:58 -0800 |
commit | f41959ccb2d9d4c722fe8fc3351401d53bcf4900 (patch) | |
tree | ef0ca22cb2a5ac4bdec9d080d8e0788a53ed496d /tensorflow/core/framework/config.proto |
TensorFlow: Initial commit of TensorFlow library.
TensorFlow is an open source software library for numerical computation
using data flow graphs.
Base CL: 107276108
Diffstat (limited to 'tensorflow/core/framework/config.proto')
-rw-r--r-- | tensorflow/core/framework/config.proto | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/tensorflow/core/framework/config.proto b/tensorflow/core/framework/config.proto new file mode 100644 index 0000000000..f0def3d6d7 --- /dev/null +++ b/tensorflow/core/framework/config.proto @@ -0,0 +1,61 @@ +syntax = "proto3"; + +package tensorflow; +// option cc_enable_arenas = true; + +message GPUOptions { + // A value between 0 and 1 that indicates what fraction of the + // available GPU memory to pre-allocate for each process. 1 means + // to pre-allocate all of the GPU memory, 0.5 means the process + // allocates ~50% of the available GPU memory. + double per_process_gpu_memory_fraction = 1; +}; + +// Session configuration parameters. +// The system picks an appropriate values for fields that are not set. +message ConfigProto { + // Map from device type name (e.g., "CPU" or "GPU" ) to maximum + // number of devices of that type to use. If a particular device + // type is not found in the map, the system picks an appropriate + // number. + map<string, int32> device_count = 1; + + // The execution of an individual op (for some op types) can be + // parallelized on a pool of intra_op_parallelism_threads. + // 0 means the system picks an appropriate number. + int32 intra_op_parallelism_threads = 2; + + // Nodes that perform blocking operations are enqueued on a pool of + // inter_op_parallelism_threads available in each process. + // + // 0 means the system picks an appropriate number. + // + // Note that the first Session created in the process sets the + // number of threads for all future sessions. + int32 inter_op_parallelism_threads = 5; + + // Assignment of Nodes to Devices is recomputed every placement_period + // steps until the system warms up (at which point the recomputation + // typically slows down automatically). + int32 placement_period = 3; + + // When any filters are present sessions will ignore all devices which do not + // match the filters. Each filter can be partially specified, e.g. "/job:ps" + // "/job:worker/replica:3", etc. + repeated string device_filters = 4; + + // Options that apply to all GPUs. + GPUOptions gpu_options = 6; + + // Whether soft placement is allowed. If allow_soft_placement is true, + // an op will be placed on CPU if + // 1. there's no GPU implementation for the OP + // or + // 2. no GPU devices are known or registered + // or + // 3. need to co-locate with reftype input(s) which are from CPU. + bool allow_soft_placement = 7; + + // Whether device placements should be logged. + bool log_device_placement = 8; +}; |