Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/data/
Ddataset_utils.cc96 const auto& optimization_options = options.optimization_options(); in DefaultOptimizationGraphRewrites() local
97 if (optimization_options.optional_apply_default_optimizations_case() != in DefaultOptimizationGraphRewrites()
99 optimization_options.apply_default_optimizations()) { in DefaultOptimizationGraphRewrites()
100 if (optimization_options.optional_map_and_batch_fusion_case() != in DefaultOptimizationGraphRewrites()
104 if (optimization_options.optional_noop_elimination_case() != in DefaultOptimizationGraphRewrites()
108 if (optimization_options.optional_map_parallelization_case() != in DefaultOptimizationGraphRewrites()
112 if (optimization_options.optional_shuffle_and_repeat_fusion_case() != in DefaultOptimizationGraphRewrites()
116 if (optimization_options.optional_parallel_batch_case() != in DefaultOptimizationGraphRewrites()
124 if (optimization_options.optional_filter_fusion_case() == in DefaultOptimizationGraphRewrites()
126 if (optimization_options.filter_fusion()) { in DefaultOptimizationGraphRewrites()
[all …]
Dcaptured_function.cc638 grappler::GrapplerItem::OptimizationOptions optimization_options; in Instantiate() local
639 optimization_options.allow_pruning_stateful_and_dataset_ops = false; in Instantiate()
656 std::move(optimization_options), std::placeholders::_6); in Instantiate()
Drewrite_utils.cc295 grappler_item->optimization_options().optimize_function_library = false; in GetGrapplerItem()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dtransform_graph_function.cc115 GraphOptimizationPassOptions optimization_options; in TransformGraphFunction() local
122 optimization_options.session_options = &session_options; in TransformGraphFunction()
123 optimization_options.graph = &graph; in TransformGraphFunction()
124 optimization_options.flib_def = func_lib_def; in TransformGraphFunction()
125 optimization_options.device_set = &device_set; in TransformGraphFunction()
126 optimization_options.is_function_graph = true; in TransformGraphFunction()
127 optimization_options.default_function_device = default_device; in TransformGraphFunction()
128 optimization_options.function_def = &fdef; in TransformGraphFunction()
131 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options)); in TransformGraphFunction()
/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/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/grappler/optimizers/
Dmeta_optimizer_test.cc97 optimization_options) { in SetOptimizationOptions()
98 optimization_options_ = optimization_options; in SetOptimizationOptions()
117 optimization_options_->insert({item.id, item.optimization_options()}); in Optimize()
626 gtl::FlatMap<string, GrapplerItem::OptimizationOptions> optimization_options; in TEST_F() local
628 &optimization_options); in TEST_F()
687 ASSERT_EQ(optimization_options.size(), 3); in TEST_F()
690 gtl::FindOrNull(optimization_options, "main"); in TEST_F()
695 gtl::FindOrNull(optimization_options, "MyMul1"); in TEST_F()
700 gtl::FindOrNull(optimization_options, "MyMul2"); in TEST_F()
835 gtl::FlatMap<string, GrapplerItem::OptimizationOptions> optimization_options; in TEST_F() local
[all …]
Dmeta_optimizer.cc1072 item.optimization_options().optimize_function_library; in OptimizeConsumeItem()
1199 func_item.optimization_options().allow_non_differentiable_rewrites = in OptimizeConsumeItem()
1214 func_item.optimization_options().allow_pruning_stateful_and_dataset_ops = in OptimizeConsumeItem()
1372 const GrapplerItem::OptimizationOptions& optimization_options, in OptimizeGraph() argument
1380 item.optimization_options() = optimization_options; in OptimizeGraph()
Dmeta_optimizer.h159 const GrapplerItem::OptimizationOptions& optimization_options,
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc613 GraphOptimizationPassOptions optimization_options; in InitBaseGraph() local
614 optimization_options.session_handle = session_handle_; in InitBaseGraph()
615 optimization_options.session_options = session_options_; in InitBaseGraph()
616 optimization_options.graph = &new_graph; in InitBaseGraph()
617 optimization_options.flib_def = flib_def_.get(); in InitBaseGraph()
618 optimization_options.device_set = device_set_; in InitBaseGraph()
621 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options)); in InitBaseGraph()
633 OptimizationPassRegistry::POST_PLACEMENT, optimization_options)); in InitBaseGraph()
898 GraphOptimizationPassOptions optimization_options; in BuildGraph() local
899 optimization_options.session_options = session_options_; in BuildGraph()
[all …]
Dprocess_function_library_runtime.cc890 GraphOptimizationPassOptions optimization_options; in InstantiateMultiDevice() local
895 optimization_options.session_options = &session_options; in InstantiateMultiDevice()
896 optimization_options.graph = &graph; in InstantiateMultiDevice()
897 optimization_options.flib_def = &data->lib_def_; in InstantiateMultiDevice()
898 optimization_options.device_set = dev_set.get(); in InstantiateMultiDevice()
899 optimization_options.is_function_graph = true; in InstantiateMultiDevice()
905 optimization_options.composite_devices = &composite_devices; in InstantiateMultiDevice()
906 optimization_options.default_function_device = default_device; in InstantiateMultiDevice()
907 optimization_options.function_def = fdef; in InstantiateMultiDevice()
908 optimization_options.shape_inference_on_tfe_dialect_import = in InstantiateMultiDevice()
[all …]
Ddirect_session.cc1725 GraphOptimizationPassOptions optimization_options; in CreateGraphs() local
1726 optimization_options.session_options = &options_; in CreateGraphs()
1727 optimization_options.flib_def = client_graph->flib_def.get(); in CreateGraphs()
1728 optimization_options.partition_graphs = outputs; in CreateGraphs()
1730 OptimizationPassRegistry::POST_PARTITIONING, optimization_options)); in CreateGraphs()
Dplacer_test.cc294 GraphOptimizationPassOptions optimization_options; in CallOptPassesAndPlace() local
296 optimization_options.graph = &graph_ptr; in CallOptPassesAndPlace()
298 optimization_options.flib_def = &flib_def; in CallOptPassesAndPlace()
299 optimization_options.device_set = &devices_; in CallOptPassesAndPlace()
300 optimization_options.session_options = &session_options; in CallOptPassesAndPlace()
302 OptimizationPassRegistry::PRE_PLACEMENT, optimization_options); in CallOptPassesAndPlace()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc183 grappler::GrapplerItem::OptimizationOptions optimization_options; in Instantiate() local
189 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in Instantiate()
193 optimization_options.optimize_function_library = false; in Instantiate()
199 func_->name(), optimization_options, std::placeholders::_6); in Instantiate()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc1308 GraphOptimizationPassOptions optimization_options; in ComputeAsync() local
1316 optimization_options.session_handle = ctx->session_handle(); in ComputeAsync()
1317 optimization_options.session_options = &session_options; in ComputeAsync()
1318 optimization_options.graph = &graph; in ComputeAsync()
1319 optimization_options.flib_def = flib_def_.get(); in ComputeAsync()
1320 optimization_options.device_set = &device_set_; in ComputeAsync()
1322 ctx, PlacementHelper(device_set_, optimization_options, func_.name()), in ComputeAsync()
1334 optimization_options.graph = nullptr; in ComputeAsync()
1335 optimization_options.device_set = nullptr; in ComputeAsync()
1336 optimization_options.partition_graphs = &subgraphs; in ComputeAsync()
[all …]
Dtpu_functional_ops.h226 const GraphOptimizationPassOptions& optimization_options,
232 const GraphOptimizationPassOptions& optimization_options, Graph* graph,
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.cc237 GraphOptimizationPassOptions optimization_options; in InlineFunctions() local
244 optimization_options.session_options = &session_options; in InlineFunctions()
245 optimization_options.graph = graph; in InlineFunctions()
246 optimization_options.flib_def = (*graph)->mutable_flib_def(); in InlineFunctions()
247 optimization_options.device_set = device_set; in InlineFunctions()
248 optimization_options.is_function_graph = false; in InlineFunctions()
251 return pass.Run(optimization_options); in InlineFunctions()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc204 GraphOptimizationPassOptions optimization_options; in InitItem() local
205 optimization_options.flib_def = item->lib_def.get(); in InitItem()
206 optimization_options.partition_graphs = &partition_graphs; in InitItem()
208 OptimizationPassRegistry::POST_PARTITIONING, optimization_options)); in InitItem()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc200 grappler::GrapplerItem::OptimizationOptions optimization_options = in InstantiateFunc() local
207 optimization_options, std::placeholders::_6); in InstantiateFunc()
/external/tensorflow/tensorflow/python/data/ops/
Doptions.py622 …pb.optimization_options.CopyFrom(self.experimental_optimization._to_proto()) # pylint: disable=pr…
637 …self.experimental_optimization._from_proto(pb.optimization_options) # pylint: disable=protected-a…
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto187 OptimizationOptions optimization_options = 3; field
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py184 expected_pb.optimization_options.CopyFrom(
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc72 optimization_options().allow_pruning_stateful_and_dataset_ops = false; in GrapplerFunctionItem()
Dfunctions_test.cc472 const auto &opts = item.optimization_options(); in TEST_F()

12