Searched refs:config_proto_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 128 ConfigProto config_proto_; member in tensorflow::MlirGraphOptimizationPassTest
145 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
160 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
174 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
190 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
209 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
226 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | custom_graph_optimizer.h | 37 config_proto_ = config_proto;
41 ConfigProto config_proto_; variable
|
| D | meta_optimizer.cc | 174 if (config_proto_.experimental().executor_type() == in LowerControlFlow()
178 if (config_proto_.experimental().use_tfrt()) return false; in LowerControlFlow()
224 config_proto_(cfg), in MetaOptimizer()
225 cfg_(*config_proto_.mutable_graph_options()->mutable_rewrite_options()) { in MetaOptimizer()
295 config_proto_.graph_options().optimizer_options().global_jit_level(); in InitializeOptimizers()
335 config_proto_, GetCustomGraphOptimizerConfig(optimizer_name))); in InitializeOptimizersByName()
362 custom_optimizer->InitWithConfig(config_proto_, &optimizer_config)); in InitializeCustomGraphOptimizers()
|
| D | meta_optimizer.h | 93 ConfigProto config_proto_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 46 config_proto_(new ConfigProto), in PartitionedCallOp()
62 config_proto_->mutable_graph_options() in PartitionedCallOp()
69 ctx, config_proto_->ParseFromString(config_proto_serialized), in PartitionedCallOp()
197 std::placeholders::_4, std::placeholders::_5, *config_proto_, in Instantiate()
|
| D | partitioned_function_ops.h | 59 std::unique_ptr<ConfigProto> config_proto_; variable
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration_test.cc | 194 &config_proto_)); in SetUp()
196 ExampleParserConfiguration config_proto_; member in tensorflow::__anonedc01a510111::ExampleParserConfigurationProtoToFeatureVectorsTest
203 config_proto_, &fixed_len_features, &var_len_features)); in TEST_F()
|