| /external/tensorflow/tensorflow/python/client/ |
| D | device_lib.i | 18 %typemap(in) const tensorflow::ConfigProto& (tensorflow::ConfigProto temp) {
47 const tensorflow::ConfigProto& config, TF_Status* out_status) { in ListDevicesWithSessionConfig()
75 tensorflow::ConfigProto session_config; in ListDevices()
96 const tensorflow::ConfigProto& config, TF_Status* out_status);
|
| D | session_clusterspec_prop_test.py | 61 config = config_pb2.ConfigProto(cluster_def=cluster_def)
76 config = config_pb2.ConfigProto(cluster_def=cluster_def)
104 config = config_pb2.ConfigProto(cluster_def=cluster_def)
120 config = config_pb2.ConfigProto(cluster_def=cluster_def)
148 config = config_pb2.ConfigProto(cluster_def=cluster_def)
183 server_config = config_pb2.ConfigProto(device_count={'CPU': 2})
191 config = config_pb2.ConfigProto(cluster_def=cluster_def)
222 config = config_pb2.ConfigProto(cluster_def=cluster_def)
265 config1 = config_pb2.ConfigProto(cluster_def=cluster_def1)
266 config2 = config_pb2.ConfigProto(cluster_def=cluster_def2)
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | copy_to_device_test.py | 57 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
82 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
106 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
156 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
181 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
211 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
244 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
268 config=config_pb2.ConfigProto(allow_soft_placement=False)):
289 config=config_pb2.ConfigProto(allow_soft_placement=False)):
[all …]
|
| D | prefetch_to_device_test.py | 53 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
103 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
154 config=config_pb2.ConfigProto(allow_soft_placement=False)):
179 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
203 config=config_pb2.ConfigProto(allow_soft_placement=False)):
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 48 config = config_pb2.ConfigProto(device_count={"CPU": 3})
59 config = config_pb2.ConfigProto(device_count={"CPU": 3})
78 config = config_pb2.ConfigProto(device_count={"CPU": 2})
97 config = config_pb2.ConfigProto(device_count={"CPU": 3})
121 config = config_pb2.ConfigProto(device_count={"CPU": 3})
149 config = config_pb2.ConfigProto(device_count={"CPU": 3})
178 config = config_pb2.ConfigProto(device_count={"CPU": 3})
207 config = config_pb2.ConfigProto(device_count={"CPU": 3})
239 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
260 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.h | 37 MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg);
80 ConfigProto config_proto_;
101 bool MetaOptimizerEnabled(const ConfigProto& cfg);
109 Status RunMetaOptimizer(const GrapplerItem& item, const ConfigProto& cfg,
134 Device* cpu_device, const ConfigProto& config_proto,
|
| D | meta_optimizer_test.cc | 141 ConfigProto config_proto; in TEST_F()
160 ConfigProto config_proto; in TEST_F()
182 ConfigProto config_proto; in TEST_F()
203 ConfigProto config_proto; in TEST_F()
220 ConfigProto config_proto; in TEST_F()
239 ConfigProto config_proto; in TEST_F()
410 ConfigProto config_proto; in TEST_F()
502 ConfigProto config_proto; in TEST_F()
615 ConfigProto config_proto; in TEST_F()
719 ConfigProto config; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer.i | 37 %typemap(in) const tensorflow::ConfigProto& (
38 tensorflow::ConfigProto temp) {
96 const tensorflow::ConfigProto& config_proto, in TF_OptimizeGraph()
129 const tensorflow::ConfigProto& config_proto,
|
| D | memory_optimizer_test.py | 54 config = config_pb2.ConfigProto()
81 config = config_pb2.ConfigProto()
138 config = config_pb2.ConfigProto()
166 config = config_pb2.ConfigProto()
196 config = config_pb2.ConfigProto()
227 return config_pb2.ConfigProto(graph_options=graph_options)
248 config=config_pb2.ConfigProto(),
300 session_config = config_pb2.ConfigProto(graph_options=graph_options)
313 config = config_pb2.ConfigProto()
|
| D | tf_optimizer_test.py | 50 config = config_pb2.ConfigProto()
75 config = config_pb2.ConfigProto()
120 config = config_pb2.ConfigProto()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.h | 34 class ConfigProto; variable
40 GrpcWorker(WorkerEnv* env, const ConfigProto& config);
62 const ConfigProto& config);
|
| D | grpc_server_lib.h | 50 const ConfigProto&, const WorkerEnv*, WorkerCacheInterface*)>
60 const ConfigProto& config)>
|
| D | grpc_master_service.cc | 50 GrpcMasterService(Master* master, const ConfigProto& default_session_config, in GrpcMasterService()
142 const ConfigProto default_session_config_;
303 Master* master, const ConfigProto& default_session_config, in NewGrpcMasterService()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.pbtxt | 1 path: "tensorflow.ConfigProto"
4 name: "ConfigProto"
10 type_name: ".tensorflow.ConfigProto.DeviceCountEntry"
106 type_name: ".tensorflow.ConfigProto.Experimental"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 35 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
56 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
96 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
119 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | predictor_factories_test.py | 48 self._export_dir, config=config_pb2.ConfigProto())
67 config=config_pb2.ConfigProto())
84 estimator, input_fn, config=config_pb2.ConfigProto())
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.h | 25 class ConfigProto; variable
36 const ConfigProto& config, const DeviceMgr* dev_mgr, in GdrCollectiveExecutorMgr()
|
| D | gdr_server_lib.cc | 58 const ConfigProto& config) { in Init()
63 [this](const ConfigProto& config, const WorkerEnv* env, in Init()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.h | 27 class ConfigProto; variable
59 std::unique_ptr<ConfigProto> config_proto_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.h | 22 class ConfigProto; variable
27 CollectiveExecutorMgr(const ConfigProto& config, const DeviceMgr* dev_mgr,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.h | 23 class ConfigProto; variable
38 const ConfigProto& config, const DeviceMgr* dev_mgr,
|
| D | collective_param_resolver_distributed.h | 21 class ConfigProto; variable
28 CollectiveParamResolverDistributed(const ConfigProto& config,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib_test.py | 94 config=config_pb2.ConfigProto(operation_timeout_in_ms=5))
120 return config_pb2.ConfigProto(rpc_options=config_pb2.RPCOptions(
185 config = config_pb2.ConfigProto(
256 sharing_config = config_pb2.ConfigProto(isolate_session_state=False)
260 isolate_config = config_pb2.ConfigProto(isolate_session_state=True)
308 sharing_config = config_pb2.ConfigProto(isolate_session_state=False)
309 isolate_config = config_pb2.ConfigProto(isolate_session_state=True)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduce_benchmark_test.py | 73 config = config_pb2.ConfigProto(
89 config = config_pb2.ConfigProto(
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | session_bundle_test.py | 94 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
122 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
132 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
|