Home
last modified time | relevance | path

Searched refs:optimization_options (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc564 GraphOptimizationPassOptions optimization_options; in InitBaseGraph() local
565 optimization_options.session_handle = session_handle_; in InitBaseGraph()
566 optimization_options.session_options = session_options_; in InitBaseGraph()
567 optimization_options.graph = &new_graph; in InitBaseGraph()
568 optimization_options.flib_def = flib_def_.get(); in InitBaseGraph()
569 optimization_options.device_set = device_set_; in InitBaseGraph()
572 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options)); in InitBaseGraph()
583 OptimizationPassRegistry::POST_PLACEMENT, optimization_options)); in InitBaseGraph()
781 GraphOptimizationPassOptions optimization_options; in BuildGraph() local
782 optimization_options.session_options = session_options_; in BuildGraph()
[all …]
Dprocess_function_library_runtime.cc581 GraphOptimizationPassOptions optimization_options; in InstantiateMultiDevice() local
586 optimization_options.session_options = &session_options; in InstantiateMultiDevice()
587 optimization_options.graph = &graph; in InstantiateMultiDevice()
588 optimization_options.flib_def = &data->overlay_lib_; in InstantiateMultiDevice()
589 optimization_options.device_set = &device_set; in InstantiateMultiDevice()
593 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options)); in InstantiateMultiDevice()
620 OptimizationPassRegistry::POST_PLACEMENT, optimization_options)); in InstantiateMultiDevice()
639 OptimizationPassRegistry::POST_REWRITE_FOR_EXEC, optimization_options)); in InstantiateMultiDevice()
Ddirect_session.cc1583 GraphOptimizationPassOptions optimization_options; in CreateGraphs() local
1584 optimization_options.session_options = &options_; in CreateGraphs()
1585 optimization_options.flib_def = client_graph->flib_def.get(); in CreateGraphs()
1586 optimization_options.partition_graphs = outputs; in CreateGraphs()
1588 OptimizationPassRegistry::POST_PARTITIONING, optimization_options)); in CreateGraphs()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
DBUILD45 "//tensorflow/python/data/experimental/ops:optimization_options",
72 "//tensorflow/python/data/experimental/ops:optimization_options",
132 "//tensorflow/python/data/experimental/ops:optimization_options",
154 "//tensorflow/python/data/experimental/ops:optimization_options",
174 "//tensorflow/python/data/experimental/ops:optimization_options",
201 "//tensorflow/python/data/experimental/ops:optimization_options",
238 "//tensorflow/python/data/experimental/ops:optimization_options",
352 "//tensorflow/python/data/experimental/ops:optimization_options",
375 "//tensorflow/python/data/experimental/ops:optimization_options",
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc96 optimization_options) { in SetOptimizationOptions()
97 optimization_options_ = optimization_options; in SetOptimizationOptions()
115 optimization_options_->insert({item.id, item.optimization_options()}); in Optimize()
610 gtl::FlatMap<string, GrapplerItem::OptimizationOptions> optimization_options; in TEST_F() local
612 &optimization_options); in TEST_F()
671 ASSERT_EQ(optimization_options.size(), 3); in TEST_F()
674 gtl::FindOrNull(optimization_options, "main"); in TEST_F()
679 gtl::FindOrNull(optimization_options, "MyMul1"); in TEST_F()
684 gtl::FindOrNull(optimization_options, "MyMul2"); in TEST_F()
795 gtl::FlatMap<string, GrapplerItem::OptimizationOptions> optimization_options; in TEST_F() local
[all …]
Dmeta_optimizer.cc568 item.optimization_options().optimize_function_library; in Optimize()
606 func_item.optimization_options().allow_non_differentiable_rewrites = in Optimize()
620 func_item.optimization_options().allow_pruning_stateful_and_dataset_ops = in Optimize()
719 const GrapplerItem::OptimizationOptions& optimization_options, in OptimizeGraph() argument
727 item.optimization_options() = optimization_options; in OptimizeGraph()
Dmeta_optimizer.h136 const GrapplerItem::OptimizationOptions& optimization_options,
Darithmetic_optimizer.cc3665 item.optimization_options().allow_non_differentiable_rewrites; in Optimize()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc157 grappler::GrapplerItem::OptimizationOptions optimization_options; in Instantiate() local
164 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in Instantiate()
168 optimization_options.optimize_function_library = false; in Instantiate()
179 func_->name(), optimization_options, std::placeholders::_6); in Instantiate()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.h117 const OptimizationOptions& optimization_options() const;
118 OptimizationOptions& optimization_options();
Dgrappler_item.cc181 const GrapplerItem::OptimizationOptions& GrapplerItem::optimization_options() in optimization_options() function in tensorflow::grappler::GrapplerItem
186 GrapplerItem::OptimizationOptions& GrapplerItem::optimization_options() { in optimization_options() function in tensorflow::grappler::GrapplerItem
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py21 from tensorflow.python.data.experimental.ops import optimization_options
88 optimization_options.OptimizationOptions())
DBUILD435 "//tensorflow/python/data/experimental/ops:optimization_options",
494 "//tensorflow/python/data/experimental/ops:optimization_options",
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc123 grappler::GrapplerItem::OptimizationOptions optimization_options; in Init() local
130 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in Init()
134 optimization_options.optimize_function_library = false; in Init()
140 optimization_options, std::placeholders::_6); in Init()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc190 GraphOptimizationPassOptions optimization_options; in InitItem() local
191 optimization_options.flib_def = item->lib_def.get(); in InitItem()
192 optimization_options.partition_graphs = &partition_graphs; in InitItem()
194 OptimizationPassRegistry::POST_PARTITIONING, optimization_options)); in InitItem()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-optimization-options.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.optimization_options.OptimizationOp…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-optimization-options.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.optimization_options.OptimizationOp…
/external/tensorflow/tensorflow/python/data/ops/
DBUILD29 "//tensorflow/python/data/experimental/ops:optimization_options",
Ddataset_ops.py31 from tensorflow.python.data.experimental.ops import optimization_options
1975 ty=optimization_options.OptimizationOptions,
1979 default_factory=optimization_options.OptimizationOptions)
/external/tensorflow/tensorflow/python/data/experimental/ops/
DBUILD252 name = "optimization_options",
253 srcs = ["optimization_options.py"],
/external/tensorflow/tensorflow/python/data/experimental/
D__init__.py105 from tensorflow.python.data.experimental.ops.optimization_options import OptimizationOptions
/external/tensorflow/tensorflow/core/kernels/data/
Dgraph_rewrite_dataset.cc173 grappler_item->optimization_options().optimize_function_library = in ApplyOptimizations()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc345 optimization_options().allow_pruning_stateful_and_dataset_ops = false; in GrapplerFunctionItem()
Dfunctions_test.cc645 const auto &opts = item.optimization_options(); in TEST_F()