Home
last modified time | relevance | path

Searched refs:optimizer_opts (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc254 OptimizerOptions* optimizer_opts = in RuntimeGraphOptimizer() local
257 optimizer_opts->set_opt_level(::tensorflow::OptimizerOptions::L1); in RuntimeGraphOptimizer()
259 optimizer_opts->set_opt_level(::tensorflow::OptimizerOptions::L0); in RuntimeGraphOptimizer()
261 optimizer_opts->set_do_function_inlining(cfg.inline_functions); in RuntimeGraphOptimizer()
267 &function_library, *optimizer_opts)); in RuntimeGraphOptimizer()
280 ::tensorflow::GraphOptimizer optimizer(*optimizer_opts); in RuntimeGraphOptimizer()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc507 GraphOptimizer::Options optimizer_opts; in OptimizeGraph() local
508 optimizer_opts.inline_multi_device_functions = true; in OptimizeGraph()
509 optimizer_opts.inline_impl_selection_group_functions = true; in OptimizeGraph()
510 optimizer_opts.inline_with_single_device_body_placer = true; in OptimizeGraph()
520 optimizer_opts.shape_map = &shape_map; in OptimizeGraph()
521 optimizer.Optimize(flr, flr->env(), flr->device(), graph, optimizer_opts); in OptimizeGraph()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc212 const auto& optimizer_opts = graph_options.optimizer_options(); in InitItem() local
213 GraphOptimizer optimizer(optimizer_opts); in InitItem()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc1336 const auto& optimizer_opts = in CreateExecutors() local
1347 func_info->flib_def.get(), optimizer_opts, thread_pools_[0].first, in CreateExecutors()
1355 GraphOptimizer optimizer(optimizer_opts); in CreateExecutors()
Dplacer_test.cc274 OptimizerOptions* optimizer_opts = graph_opts->mutable_optimizer_options(); in CallOptPassesAndPlace() local
275 optimizer_opts->set_opt_level(OptimizerOptions::L0); in CallOptPassesAndPlace()
276 optimizer_opts->set_global_jit_level(OptimizerOptions::OFF); in CallOptPassesAndPlace()