Home
last modified time | relevance | path

Searched refs:ConfigProto (Results 1 – 25 of 202) sorted by relevance

123456789

/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_bridge_rollout_policy.cc22 static ConfigProto::Experimental::MlirBridgeRollout GetUserRequest( in GetUserRequest()
23 absl::optional<ConfigProto> config_proto) { in GetUserRequest()
29 ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_UNSPECIFIED) { in GetUserRequest()
38 return ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_UNSPECIFIED; in GetUserRequest()
48 return ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_ENABLED; in GetUserRequest()
54 const tensorflow::Graph& graph, absl::optional<ConfigProto> config_proto, in GetMlirBridgeRolloutPolicy()
57 case ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_ENABLED: in GetMlirBridgeRolloutPolicy()
59 case ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_DISABLED: in GetMlirBridgeRolloutPolicy()
Dmlir_graph_optimization_pass.h67 const DeviceSet* device_set, const ConfigProto& config_proto,
70 virtual Status Run(const ConfigProto& config_proto, mlir::ModuleOp module,
134 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto,
161 const ConfigProto& config_proto,
Dmlir_graph_optimization_pass_test.cc37 const ConfigProto& config_proto,
39 MOCK_METHOD3(Run, Status(const ConfigProto& config_proto,
51 const ConfigProto& config_proto,
56 Status Run(const ConfigProto& config_proto, mlir::ModuleOp module, in Run()
128 ConfigProto config_proto_;
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.h37 MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg);
93 ConfigProto config_proto_;
115 bool MetaOptimizerEnabled(const ConfigProto& cfg);
123 Status RunMetaOptimizer(GrapplerItem&& item, const ConfigProto& cfg,
148 Device* cpu_device, const ConfigProto& config_proto,
Dmeta_optimizer_test.cc146 ConfigProto config_proto; in TEST_F()
165 ConfigProto config_proto; in TEST_F()
187 ConfigProto config_proto; in TEST_F()
208 ConfigProto config_proto; in TEST_F()
225 ConfigProto config_proto; in TEST_F()
244 ConfigProto config_proto; in TEST_F()
408 ConfigProto config_proto; in TEST_F()
500 ConfigProto config_proto; in TEST_F()
613 ConfigProto config_proto; in TEST_F()
719 ConfigProto config; in TEST_F()
[all …]
Dcustom_graph_optimizer.h35 const ConfigProto& config_proto,
41 ConfigProto config_proto_;
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py60 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
85 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
109 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
134 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
159 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
184 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
214 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
247 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
309 config=config_pb2.ConfigProto(allow_soft_placement=False)):
333 config=config_pb2.ConfigProto(allow_soft_placement=False)):
[all …]
Dprefetch_to_device_test.py56 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
81 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
105 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
132 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
181 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
205 config=config_pb2.ConfigProto(allow_soft_placement=False)):
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_bridge_pass.h34 const ConfigProto& config_proto,
39 Status Run(const ConfigProto& config_proto, mlir::ModuleOp module,
50 bool IsEnabled(const DeviceSet* device_set, const ConfigProto& config_proto,
Dmlir_bridge_pass.cc87 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
115 Status MlirBridgePass::Run(const ConfigProto& config_proto, in Run()
142 const ConfigProto& config_proto, in IsEnabled()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgraph_optimization_pass.cc30 using ConfigProto = ::tensorflow::ConfigProto; typedef
34 Status MlirGraphOptimizationPass::Run(const ConfigProto& config_proto, in Run()
Dgraph_optimization_pass.h32 const ::tensorflow::ConfigProto& config_proto, in GetPassState()
39 ::tensorflow::Status Run(const ::tensorflow::ConfigProto& config_proto,
/external/tensorflow/tensorflow/python/client/
Dsession_clusterspec_prop_test.py55 config = config_pb2.ConfigProto(cluster_def=cluster_def)
70 config = config_pb2.ConfigProto(cluster_def=cluster_def)
98 config = config_pb2.ConfigProto(cluster_def=cluster_def)
114 config = config_pb2.ConfigProto(cluster_def=cluster_def)
142 config = config_pb2.ConfigProto(cluster_def=cluster_def)
177 server_config = config_pb2.ConfigProto(device_count={'CPU': 2})
185 config = config_pb2.ConfigProto(cluster_def=cluster_def)
216 config = config_pb2.ConfigProto(cluster_def=cluster_def)
259 config1 = config_pb2.ConfigProto(cluster_def=cluster_def1)
260 config2 = config_pb2.ConfigProto(cluster_def=cluster_def2)
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.h34 class ConfigProto; variable
41 GrpcWorker(WorkerEnv* env, const ConfigProto& config);
71 const ConfigProto& config);
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmulti_device_iterator_test.py55 config = config_pb2.ConfigProto(device_count={"CPU": 3})
72 config = config_pb2.ConfigProto(device_count={"CPU": 3})
91 config = config_pb2.ConfigProto(device_count={"CPU": 2})
110 config = config_pb2.ConfigProto(device_count={"CPU": 3})
134 config = config_pb2.ConfigProto(device_count={"CPU": 3})
162 config = config_pb2.ConfigProto(device_count={"CPU": 3})
191 config = config_pb2.ConfigProto(device_count={"CPU": 3})
220 config = config_pb2.ConfigProto(device_count={"CPU": 3})
254 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
275 config = config_pb2.ConfigProto(device_count={"CPU": 2, "GPU": 1})
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py54 config = config_pb2.ConfigProto()
82 config = config_pb2.ConfigProto()
139 config = config_pb2.ConfigProto()
167 config = config_pb2.ConfigProto()
197 config = config_pb2.ConfigProto()
228 return config_pb2.ConfigProto(graph_options=graph_options)
249 config=config_pb2.ConfigProto(),
301 session_config = config_pb2.ConfigProto(graph_options=graph_options)
315 config = config_pb2.ConfigProto()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-config-proto.pbtxt1 path: "tensorflow.ConfigProto"
4 name: "ConfigProto"
10 type_name: ".tensorflow.ConfigProto.DeviceCountEntry"
112 type_name: ".tensorflow.ConfigProto.Experimental"
212 type_name: ".tensorflow.ConfigProto.Experimental.MlirBridgeRollout"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/
Denable_mlir_bridge.md7 In tf.ConfigProto.Experimental, there is a knob controlling whether the new TPU
11 session_config = tf.ConfigProto(
13 experimental=tf.ConfigProto.Experimental(
/external/tensorflow/tensorflow/compiler/jit/
Dflags.cc245 ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_UNSPECIFIED; in AllocateAndParseFlags()
249 ? ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED in AllocateAndParseFlags()
250 : ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_ENABLED; in AllocateAndParseFlags()
253 ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_DISABLED; in AllocateAndParseFlags()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dgraph_decompose_pass.h37 const DeviceSet* device_set, const ConfigProto& config_proto,
42 Status Run(const ConfigProto& config_proto, mlir::ModuleOp module,
Dgraph_decompose_pass.cc34 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
41 Status GraphDecomposePass::Run(const ConfigProto& config_proto, in Run()
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_test.py65 config = config_pb2.ConfigProto(device_count={device_type: group_size})
98 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
188 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
249 config = config_pb2.ConfigProto(device_count={'CPU': group_size})
307 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
336 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
378 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
402 config=config_pb2.ConfigProto(device_count={'CPU': 2})) as sess:
424 config=config_pb2.ConfigProto(
528 cfg = config_pb2.ConfigProto(device_count={'CPU': group_size},
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_optimization_registry_pass_failure_test.cc31 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 ConfigProto config_proto; in TEST()
Dfunction_optimization_registry_test.cc32 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
49 ConfigProto config_proto; in TEST()
Dfunction_optimization_registration_test.cc30 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 ConfigProto config_proto; in TEST()

123456789