Home
last modified time | relevance | path

Searched defs:config_proto (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_bridge_rollout_policy.cc23 absl::optional<ConfigProto> config_proto) { in GetUserRequest()
54 const tensorflow::Graph& graph, absl::optional<ConfigProto> config_proto, in GetMlirBridgeRolloutPolicy()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
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/compiler/tf2xla/
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/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() local
Dfunction_optimization_registration_test.cc30 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 ConfigProto config_proto; in TEST() local
Dfunction_optimization_registry_test.cc32 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
49 ConfigProto config_proto; in TEST() local
Dfunction_optimization_registry_no_pass_test.cc30 ConfigProto config_proto; in TEST() local
Dfunction_optimization_registry.cc27 const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc146 ConfigProto config_proto; in TEST_F() local
165 ConfigProto config_proto; in TEST_F() local
187 ConfigProto config_proto; in TEST_F() local
208 ConfigProto config_proto; in TEST_F() local
225 ConfigProto config_proto; in TEST_F() local
244 ConfigProto config_proto; in TEST_F() local
408 ConfigProto config_proto; in TEST_F() local
500 ConfigProto config_proto; in TEST_F() local
613 ConfigProto config_proto; in TEST_F() local
785 ConfigProto config_proto; in TEST_F() local
[all …]
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py42 def get_accelerator_devices(master, config_proto): argument
137 config_proto=None): argument
395 config_proto=None): argument
618 config_proto=None): argument
Dslurm_cluster_resolver.py398 config_proto=None): argument
Dtfconfig_cluster_resolver.py147 config_proto=None): argument
Dsagemaker_cluster_resolver.py154 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgraph_optimization_pass.h30 ::tensorflow::MlirOptimizationPassState GetPassState( in GetPassState()
Dgraph_optimization_pass.cc34 Status MlirGraphOptimizationPass::Run(const ConfigProto& config_proto, in Run()
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.py27 def OptimizeGraph(config_proto, argument
Dtf_optimizer_wrapper.cc68 tensorflow::ConfigProto config_proto; in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py29 def grappler_optimize(graph, fetches=None, config_proto=None): argument
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py1419 def config_proto(self): member in TPUEmbedding
2017 table_config, table_variables, config_proto): argument
2031 table_config, table_variables, config_proto): argument
2108 table_config, table_variables, config_proto): argument
2190 table_config, table_variables, config_proto): argument
2288 table_config, table_variables, config_proto): argument
2385 table_config, table_variables, config_proto): argument
2473 table_config, table_variables, config_proto): argument
2553 table_config, table_variables, config_proto): argument
2642 table_config, table_variables, config_proto): argument
[all …]
/external/perfetto/src/tracing/consumer_api_deprecated/
Dconsumer_api_deprecated.cc321 TraceConfig config_proto; in Create() local
400 PERFETTO_EXPORTED_API Handle Create(const void* config_proto, in Create()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator.cc102 absl::optional<ConfigProto> config_proto; in CreateXlaKernel() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.cc136 ConfigProto config_proto; in RunGrappler() local
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc164 const ExampleParserConfiguration& config_proto, in ExampleParserConfigurationProtoToFeatureVectors()
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/
Dtpu_cluster_resolver.py347 config_proto=None): argument
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc173 const xrt::XRTExecutionConfig& config_proto, in AllocateOutputTensors()
299 xrt::XRTExecutionConfig config_proto; in DoWork() local

12