Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc141 ConfigProto config_proto; in TEST_F() local
143 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
147 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F()
160 ConfigProto config_proto; in TEST_F() local
162 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
168 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F()
182 ConfigProto config_proto; in TEST_F() local
184 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
190 MetaOptimizer optimizer(nullptr, config_proto); in TEST_F()
203 ConfigProto config_proto; in TEST_F() local
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py26 def grappler_optimize(graph, fetches=None, config_proto=None): argument
40 if config_proto is None:
41 config_proto = config_pb2.ConfigProto()
42 config_proto.graph_options.rewrite_options.min_graph_nodes = -1
47 return tf_optimizer.OptimizeGraph(config_proto, metagraph)
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.py28 def OptimizeGraph(config_proto, argument
34 if not isinstance(config_proto, config_pb2.ConfigProto):
36 type(config_proto))
41 config_proto.SerializeToString(),
Dtf_optimizer.i96 const tensorflow::ConfigProto& config_proto, in TF_OptimizeGraph() argument
112 tensorflow::grappler::MetaOptimizer optimizer(cpu_device, config_proto); in TF_OptimizeGraph()
129 const tensorflow::ConfigProto& config_proto,
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py44 def get_accelerator_devices(master, config_proto): argument
60 with session.Session(master, config=config_proto) as s:
128 config_proto=None): argument
151 devices = get_accelerator_devices(master, config_proto)
254 config_proto=None): argument
268 del task_type, task_id, config_proto
446 config_proto=None): argument
448 task_type, task_id, config_proto)
Dtfconfig_cluster_resolver.py118 config_proto=None): argument
122 task_type, task_id, config_proto)
Dslurm_cluster_resolver.py223 config_proto=None): argument
225 del task_type, task_id, config_proto
Dtpu_cluster_resolver.py459 config_proto=None): argument
481 get_accelerator_devices(self.master(), config_proto=config_proto))
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc151 xrt::XRTExecutionConfig config_proto; in DoWork() local
153 config_proto.ParseFromString(execution_config.scalar<string>()())); in DoWork()
155 int core_index_in_replica = config_proto.core_index_in_replica(); in DoWork()
157 bool release_inputs = config_proto.release_input_handles(); in DoWork()
158 bool release_compilation = config_proto.release_compilation_handle(); in DoWork()
196 int rng_seed = config_proto.rng_seed(); in DoWork()
250 if (config_proto.return_exploded_tuple() && in DoWork()
/external/perfetto/src/tracing/api_impl/
Dconsumer_api.cc322 perfetto::protos::TraceConfig config_proto; in Create() local
323 bool parsed = config_proto.ParseFromArray(config_proto_buf, in Create()
330 if (!config_proto.duration_ms()) { in Create()
338 callback_arg, config_proto); in Create()
402 PERFETTO_EXPORTED_API Handle Create(const void* config_proto, in Create() argument
406 return TracingController::GetInstance()->Create(config_proto, config_len, in Create()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py426 def config_proto(self): member in TPUEmbedding
454 config_proto = elc.TPUEmbeddingConfiguration()
456 table_descriptor = config_proto.table_descriptor.add()
474 config_proto.mode = self._mode
475 config_proto.batch_size_per_tensor_core = self._batch_size_per_core
476 config_proto.num_hosts = self._num_hosts
477 config_proto.num_tensor_cores = self._num_cores
478 config_proto.sharding_strategy = elc.TPUEmbeddingConfiguration.DIV_DEFAULT
479 config_proto.pipeline_execution_with_tensor_core = (
482 return config_proto
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py223 config_proto = config_pb2.ConfigProto()
224 config_proto = self._update_config_proto(config_proto)
227 default_session_config=config_proto,
386 def _update_config_proto(self, config_proto): argument
387 updated_config = copy.deepcopy(config_proto)
/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/src/tracing/test/
Dtracing_integration_test.cc322 protos::TraceConfig config_proto; in TEST_F() local
323 trace_config.ToProto(&config_proto); in TEST_F()
325 static_cast<size_t>(config_proto.ByteSize())); in TEST_F()
326 config_proto.SerializeWithCachedSizesToArray( in TEST_F()
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.h50 const ExampleParserConfiguration& config_proto,
Dexample_parser_configuration.cc163 const ExampleParserConfiguration& config_proto, in ExampleParserConfigurationProtoToFeatureVectors() argument
166 const auto& feature_map = config_proto.feature_map(); in ExampleParserConfigurationProtoToFeatureVectors()
/external/perfetto/include/perfetto/public/
Dconsumer_api.h116 Handle Create(const void* config_proto,
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py822 args=args, Tout=tout, f=f, config_proto=config,
826 args=args, Tout=tout, f=f, config_proto=config,
846 config_proto = attr_value_pb2.AttrValue(
862 "config_proto": config_proto,
/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…
Dtensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt24 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt28 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…
Dtensorflow.distribute.cluster_resolver.-union-resolver.pbtxt36 …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…
Dtensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt28 …argspec: "args=[\'self\', \'task_type\', \'task_id\', \'config_proto\'], varargs=None, keywords=No…

123