| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer_test.cc | 140 ConfigProto config_proto; in TEST_F() local
159 ConfigProto config_proto; in TEST_F() local
181 ConfigProto config_proto; in TEST_F() local
203 ConfigProto config_proto; in TEST_F() local
226 ConfigProto config_proto; in TEST_F() local
243 ConfigProto config_proto; in TEST_F() local
262 ConfigProto config_proto; in TEST_F() local
426 ConfigProto config_proto; in TEST_F() local
518 ConfigProto config_proto; in TEST_F() local
631 ConfigProto config_proto; in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tf2xla/ |
| D | mlir_bridge_rollout_policy.cc | 25 std::optional<ConfigProto> config_proto, in GetMlirBridgeRolloutPolicy()
42 std::optional<ConfigProto> config_proto, in LogGraphFeatures()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_optimization_registry_test.cc | 32 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
49 ConfigProto config_proto; in TEST() local
|
| D | function_optimization_registration_test.cc | 30 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 ConfigProto config_proto; in TEST() local
|
| D | function_optimization_registry_pass_failure_test.cc | 31 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 ConfigProto config_proto; in TEST() local
|
| D | function_optimization_registry_no_pass_test.cc | 30 ConfigProto config_proto; in TEST() local
|
| D | function_optimization_registry.cc | 29 const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | graph_decompose_pass.cc | 34 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
43 const ConfigProto& config_proto, mlir::ModuleOp module, const Graph& graph, in Run()
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver.py | 38 def get_accelerator_devices(master, config_proto): argument
133 config_proto=None): argument
391 config_proto=None): argument
614 config_proto=None): argument
|
| D | slurm_cluster_resolver.py | 394 config_proto=None): argument
|
| D | tfconfig_cluster_resolver.py | 143 config_proto=None): argument
|
| D | sagemaker_cluster_resolver.py | 150 def num_accelerators(self, task_type=None, task_id=None, config_proto=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.cc | 167 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
216 Status MlirBridgePass::Run(const ConfigProto& config_proto, in Run()
266 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | graph_optimization_pass.h | 30 ::tensorflow::MlirOptimizationPassState GetPassState( in GetPassState()
|
| D | graph_optimization_pass.cc | 36 const ConfigProto& config_proto, ModuleOp module, const Graph& graph, in Run()
|
| D | constant_fold.cc | 149 tensorflow::ConfigProto config_proto; in ConstantFoldFallbackHook() local
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer.py | 27 def OptimizeGraph(config_proto, argument
|
| D | tf_optimizer_wrapper.cc | 70 tensorflow::ConfigProto config_proto; in PYBIND11_MODULE() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | test_util.py | 25 def grappler_optimize(graph, fetches=None, config_proto=None): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1500 def config_proto(self): member in TPUEmbedding
2058 table_config, table_variables, config_proto): argument
2072 table_config, table_variables, config_proto): argument
2161 table_config, table_variables, config_proto): argument
2251 table_config, table_variables, config_proto): argument
2333 table_config, table_variables, config_proto): argument
2431 table_config, table_variables, config_proto): argument
2528 table_config, table_variables, config_proto): argument
2616 table_config, table_variables, config_proto): argument
2696 table_config, table_variables, config_proto): argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | trt_convert_api.cc | 67 const ConfigProto& config_proto, grappler::Cluster* cluster, in RunGrappler()
160 ConfigProto config_proto; in RunTfTrt() local
431 ConfigProto config_proto; in InlineFunctions() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/inference/ |
| D | batch_op_rewriter_test.cc | 170 ConfigProto config_proto; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | flags.cc | 399 std::optional<const ConfigProto> config_proto) { in GetMlirBridgeRolloutState()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | transform_graph_function.cc | 146 ConfigProto config_proto; in TransformGraphFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 73 Status Run(const ConfigProto& config_proto, mlir::ModuleOp module, in Run()
|