Home
last modified time | relevance | path

Searched refs:config_proto (Results 1 – 25 of 90) sorted by relevance

1234

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc143 ConfigProto config_proto; in TEST_F() local
145 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
149 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F()
162 ConfigProto config_proto; in TEST_F() local
164 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
170 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F()
184 ConfigProto config_proto; in TEST_F() local
186 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
192 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F()
205 ConfigProto config_proto; in TEST_F() local
[all …]
Dcustom_graph_optimizer.h35 const ConfigProto& config_proto,
37 config_proto_ = config_proto;
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py29 def grappler_optimize(graph, fetches=None, config_proto=None): argument
43 if config_proto is None:
44 config_proto = config_pb2.ConfigProto()
45 config_proto.graph_options.rewrite_options.min_graph_nodes = -1
50 return tf_optimizer.OptimizeGraph(config_proto, metagraph)
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.py27 def OptimizeGraph(config_proto, argument
50 if not isinstance(config_proto, config_pb2.ConfigProto):
52 type(config_proto))
56 config_proto.SerializeToString(),
Dtf_optimizer_wrapper.cc68 tensorflow::ConfigProto config_proto; in PYBIND11_MODULE() local
69 if (!config_proto.ParseFromString(serialized_config_proto)) { in PYBIND11_MODULE()
95 tensorflow::grappler::MetaOptimizer optimizer(cpu_device, config_proto); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py42 def get_accelerator_devices(master, config_proto): argument
54 with session.Session(master, config=config_proto) as s:
122 config_proto=None): argument
145 devices = get_accelerator_devices(master, config_proto)
248 config_proto=None): argument
262 del task_type, task_id, config_proto
440 config_proto=None): argument
442 task_type, task_id, config_proto)
Dtpu_cluster_resolver.py261 config_proto=None): argument
284 self.master(), config_proto=config_proto))
Dtfconfig_cluster_resolver.py123 config_proto=None): argument
127 task_type, task_id, config_proto)
Dslurm_cluster_resolver.py223 config_proto=None): argument
225 del task_type, task_id, config_proto
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py756 def config_proto(self): member in TPUEmbedding
784 config_proto = elc.TPUEmbeddingConfiguration()
786 table_descriptor = config_proto.table_descriptor.add()
822 config_proto.mode = self._mode
823 config_proto.batch_size_per_tensor_core = self._batch_size_per_core
824 config_proto.num_hosts = self._num_hosts
825 config_proto.num_tensor_cores = self._num_cores
826 config_proto.sharding_strategy = (
830 config_proto.pipeline_execution_with_tensor_core = (
833 return config_proto
[all …]
Dtpu_system_metadata.py174 config_proto = config_pb2.ConfigProto(
176 return config_proto
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dinlining.mlir14 …%result = "tf.StatefulPartitionedCall"() {config = "", config_proto = "", executor_type = "", f = …
31 …%result = "tf.PartitionedCall"(%arg) {config = "", config_proto = "", executor_type = "", f = @inl…
51 …%result = "tf.StatefulPartitionedCall"() {config = "", config_proto = "", executor_type = "", f = …
55 …%result_2 = "tf.StatefulPartitionedCall"() {config = "", config_proto = "", executor_type = "", f …
/external/perfetto/src/tracing/consumer_api_deprecated/
Dconsumer_api_deprecated.cc321 TraceConfig config_proto; in Create() local
322 bool parsed = config_proto.ParseFromArray(config_proto_buf, config_len); in Create()
328 if (!config_proto.duration_ms()) { in Create()
336 callback_arg, config_proto); in Create()
400 PERFETTO_EXPORTED_API Handle Create(const void* config_proto, in Create() argument
404 return TracingController::GetInstance()->Create(config_proto, config_len, in Create()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc264 xrt::XRTExecutionConfig config_proto; in DoWork() local
266 config_proto.ParseFromString(execution_config.scalar<tstring>()())); in DoWork()
268 int core_index_in_replica = config_proto.core_index_in_replica(); in DoWork()
270 bool release_inputs = config_proto.release_input_handles(); in DoWork()
271 bool release_compilation = config_proto.release_compilation_handle(); in DoWork()
281 int rng_seed = config_proto.rng_seed(); in DoWork()
309 config_proto.return_exploded_tuple()); in DoWork()
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py283 config_proto = config_pb2.ConfigProto()
284 config_proto = self._update_config_proto(config_proto)
292 default_session_config=config_proto,
470 def _update_config_proto(self, config_proto): argument
471 updated_config = copy.deepcopy(config_proto)
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.h80 const ConfigProto& config_proto, int64 collective_graph_key,
184 const ConfigProto& config_proto, int64 collective_graph_key,
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.h50 const ExampleParserConfiguration& config_proto,
Dexample_parser_configuration.cc164 const ExampleParserConfiguration& config_proto, in ExampleParserConfigurationProtoToFeatureVectors() argument
167 const auto& feature_map = config_proto.feature_map(); in ExampleParserConfigurationProtoToFeatureVectors()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py243 config_proto = config_pb2.ConfigProto()
244 config_proto.gpu_options.allow_growth = True
248 config=RunConfig(session_config=config_proto))
/external/perfetto/include/perfetto/public/
Dconsumer_api.h116 Handle Create(const void* config_proto,
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc157 if (!options.config_proto.ParseFromString(config_it->second.s())) { in InstantiateFunc()
180 options.config_proto, function_def->signature().name(), in InstantiateFunc()
191 options.config_proto.mutable_graph_options() in InstantiateFunc()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.cluster_resolver.-cluster-resolver.pbtxt22 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.cluster_resolver.-cluster-resolver.pbtxt22 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dfunction-control-ret.mlir6 …sland wraps "tf.PartitionedCall"() {Tin = [], Tout = [], config = "", config_proto = "", device = …

1234