Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc35 GrapplerItem::OptimizationOptions optimization_options; in CreateOptOptionsForEager() local
41 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in CreateOptOptionsForEager()
43 optimization_options.is_eager_mode = true; in CreateOptOptionsForEager()
47 optimization_options.optimize_function_library = false; in CreateOptOptionsForEager()
49 return optimization_options; in CreateOptOptionsForEager()
219 const GrapplerItem::OptimizationOptions& GrapplerItem::optimization_options() in optimization_options() function in tensorflow::grappler::GrapplerItem
224 GrapplerItem::OptimizationOptions& GrapplerItem::optimization_options() { in optimization_options() function in tensorflow::grappler::GrapplerItem
Dgrappler_item.h121 const OptimizationOptions& optimization_options() const;
122 OptimizationOptions& optimization_options();
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
DBUILD63 "//tensorflow/python/data/experimental/ops:optimization_options",
88 "//tensorflow/python/data/experimental/ops:optimization_options",
110 "//tensorflow/python/data/experimental/ops:optimization_options",
136 "//tensorflow/python/data/experimental/ops:optimization_options",
175 "//tensorflow/python/data/experimental/ops:optimization_options",
196 "//tensorflow/python/data/experimental/ops:optimization_options",
215 "//tensorflow/python/data/experimental/ops:optimization_options",
241 "//tensorflow/python/data/experimental/ops:optimization_options",
278 "//tensorflow/python/data/experimental/ops:optimization_options",
337 "//tensorflow/python/data/experimental/ops:optimization_options",
/external/OpenCL-CTS/test_conformance/compiler/
Dtest_build_options.cpp53 const char *optimization_options[] = { variable
96 for(size_t i = 0; i < sizeof(optimization_options) / (sizeof(char*)); i++) { in test_options_build_optimizations()
99 …e_kernel_helper_create_program(context, &program, 1, options_test_kernel, optimization_options[i]); in test_options_build_optimizations()
107 log_info("Testing optimization option '%s'\n", optimization_options[i]); in test_options_build_optimizations()
108 error = clBuildProgram( program, 1, &deviceID, optimization_options[i], NULL, NULL ); in test_options_build_optimizations()
116 … log_info("Building with optimization option '%s' failed to compile!\n", optimization_options[i]); in test_options_build_optimizations()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc610 GraphOptimizationPassOptions optimization_options; in InitBaseGraph() local
611 optimization_options.session_handle = session_handle_; in InitBaseGraph()
612 optimization_options.session_options = session_options_; in InitBaseGraph()
613 optimization_options.graph = &new_graph; in InitBaseGraph()
614 optimization_options.flib_def = flib_def_.get(); in InitBaseGraph()
615 optimization_options.device_set = device_set_; in InitBaseGraph()
618 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options)); in InitBaseGraph()
630 OptimizationPassRegistry::POST_PLACEMENT, optimization_options)); in InitBaseGraph()
892 GraphOptimizationPassOptions optimization_options; in BuildGraph() local
893 optimization_options.session_options = session_options_; in BuildGraph()
[all …]
Dprocess_function_library_runtime.cc785 GraphOptimizationPassOptions optimization_options; in InstantiateMultiDevice() local
790 optimization_options.session_options = &session_options; in InstantiateMultiDevice()
791 optimization_options.graph = &graph; in InstantiateMultiDevice()
792 optimization_options.flib_def = &data->lib_def_; in InstantiateMultiDevice()
793 optimization_options.device_set = dev_set.get(); in InstantiateMultiDevice()
794 optimization_options.is_function_graph = true; in InstantiateMultiDevice()
799 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options)); in InstantiateMultiDevice()
805 Placer placer(graph.get(), function_name, optimization_options.flib_def, in InstantiateMultiDevice()
814 OptimizationPassRegistry::POST_PLACEMENT, optimization_options)); in InstantiateMultiDevice()
835 OptimizationPassRegistry::POST_REWRITE_FOR_EXEC, optimization_options)); in InstantiateMultiDevice()
[all …]
Ddirect_session.cc1708 GraphOptimizationPassOptions optimization_options; in CreateGraphs() local
1709 optimization_options.session_options = &options_; in CreateGraphs()
1710 optimization_options.flib_def = client_graph->flib_def.get(); in CreateGraphs()
1711 optimization_options.partition_graphs = outputs; in CreateGraphs()
1713 OptimizationPassRegistry::POST_PARTITIONING, optimization_options)); in CreateGraphs()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc100 optimization_options) { in SetOptimizationOptions()
101 optimization_options_ = optimization_options; in SetOptimizationOptions()
120 optimization_options_->insert({item.id, item.optimization_options()}); in Optimize()
608 gtl::FlatMap<string, GrapplerItem::OptimizationOptions> optimization_options; in TEST_F() local
610 &optimization_options); in TEST_F()
669 ASSERT_EQ(optimization_options.size(), 3); in TEST_F()
672 gtl::FindOrNull(optimization_options, "main"); in TEST_F()
677 gtl::FindOrNull(optimization_options, "MyMul1"); in TEST_F()
682 gtl::FindOrNull(optimization_options, "MyMul2"); in TEST_F()
820 gtl::FlatMap<string, GrapplerItem::OptimizationOptions> optimization_options; in TEST_F() local
[all …]
Dmeta_optimizer.cc709 item.optimization_options().optimize_function_library; in OptimizeConsumeItem()
832 func_item.optimization_options().allow_non_differentiable_rewrites = in OptimizeConsumeItem()
847 func_item.optimization_options().allow_pruning_stateful_and_dataset_ops = in OptimizeConsumeItem()
978 const GrapplerItem::OptimizationOptions& optimization_options, in OptimizeGraph() argument
986 item.optimization_options() = optimization_options; in OptimizeGraph()
Dmeta_optimizer.h150 const GrapplerItem::OptimizationOptions& optimization_options,
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc182 grappler::GrapplerItem::OptimizationOptions optimization_options; in Instantiate() local
188 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in Instantiate()
192 optimization_options.optimize_function_library = false; in Instantiate()
198 func_->name(), optimization_options, std::placeholders::_6); in Instantiate()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py28 from tensorflow.python.data.experimental.ops import optimization_options
108 optimization_options.OptimizationOptions())
185 expected_pb.optimization_options.CopyFrom(
187 expected_pb.optimization_options.map_vectorization.CopyFrom(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-map-vectorization-options.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.optimization_options.MapVectorizati…
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.-map-vectorization-options.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.optimization_options.MapVectorizati…
Dtensorflow.data.experimental.-optimization-options.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.optimization_options.OptimizationOp…
/external/tensorflow/tensorflow/python/data/experimental/
D__init__.py125 from tensorflow.python.data.experimental.ops.optimization_options import MapVectorizationOptions
126 from tensorflow.python.data.experimental.ops.optimization_options import OptimizationOptions
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc202 GraphOptimizationPassOptions optimization_options; in InitItem() local
203 optimization_options.flib_def = item->lib_def.get(); in InitItem()
204 optimization_options.partition_graphs = &partition_graphs; in InitItem()
206 OptimizationPassRegistry::POST_PARTITIONING, optimization_options)); in InitItem()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc194 grappler::GrapplerItem::OptimizationOptions optimization_options = in InstantiateFunc() local
201 optimization_options, std::placeholders::_6); in InstantiateFunc()
/external/tensorflow/tensorflow/python/data/ops/
DBUILD31 "//tensorflow/python/data/experimental/ops:optimization_options",
Ddataset_ops.py35 from tensorflow.python.data.experimental.ops import optimization_options
3013 ty=optimization_options.OptimizationOptions,
3017 default_factory=optimization_options.OptimizationOptions)
3063 …pb.optimization_options.CopyFrom(self.experimental_optimization._to_proto()) # pylint: disable=pr…
3077 …self.experimental_optimization._from_proto(pb.optimization_options) # pylint: disable=protected-a…
3096 …result = optimization_options.OptimizationOptions()._graph_rewrites() # pylint: disable=protected…
3136 …return optimization_options.OptimizationOptions()._autotune_settings() # pylint: disable=protecte…
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doptimize_dataset_test.py29 from tensorflow.python.data.experimental.ops import optimization_options
592 optimization_options._AutotuneAlgorithm.GRADIENT_DESCENT)
598 optimization_options._AutotuneAlgorithm.HILL_CLIMB)
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc124 grappler_item->optimization_options().optimize_function_library = false; in ApplyRewrites()
Dcaptured_function.cc682 grappler::GrapplerItem::OptimizationOptions optimization_options; in Instantiate() local
683 optimization_options.allow_pruning_stateful_and_dataset_ops = false; in Instantiate()
700 std::move(optimization_options), std::placeholders::_6); in Instantiate()
/external/tensorflow/tensorflow/python/data/experimental/ops/
DBUILD235 name = "optimization_options",
236 srcs = ["optimization_options.py"],

12