Home
last modified time | relevance | path

Searched refs:autotune (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dautotune_benchmark.py31 def _run_benchmark(self, dataset, autotune, autotune_buffers, argument
35 options.experimental_optimization.autotune = autotune
47 name=benchmark_label + (autotune_string if autotune else ""))
51 a = self._benchmark_batch(autotune=False)
52 b = self._benchmark_batch(autotune=True, autotune_buffers=False)
53 c = self._benchmark_batch(autotune=True, autotune_buffers=True)
58 def _benchmark_batch(self, autotune, autotune_buffers=False): argument
68 autotune,
74 a = self._benchmark_map(autotune=False)
75 b = self._benchmark_map(autotune=True, autotune_buffers=False)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Doptimization_options.py111 autotune = options.create_option( variable in OptimizationOptions
241 autotune = True
252 if self.autotune is False: # pylint: disable=g-bool-id-comparison
253 autotune = False
259 return autotune, algorithm, cpu_budget, ram_budget
312 …if self.autotune is not False and autotune_buffers is True: # pylint: disable=g-bool-id-comparison
321 if self.autotune is False: # pylint: disable=g-bool-id-comparison
327 def _graph_rewrite_configs(self, autotune): argument
339 if autotune is False: # pylint: disable=g-bool-id-comparison
352 if self.autotune is not None:
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Ddisable_prefetch_legacy_autotune_test.cc35 bool autotune) { in OptimizeWithDisablePrefetchLegacyAutotune() argument
38 if (autotune) { in OptimizeWithDisablePrefetchLegacyAutotune()
50 const bool autotune = GetParam(); in TEST_P() local
69 OptimizeWithDisablePrefetchLegacyAutotune(item, &output, autotune)); in TEST_P()
73 EXPECT_EQ(prefetch_node1.attr().at("legacy_autotune").b(), !autotune); in TEST_P()
79 if (autotune) { in TEST_P()
Denable_gradient_descent_test.cc32 GraphDef *output, bool autotune) { in OptimizeWithEnableGradientDescent() argument
35 if (autotune) { in OptimizeWithEnableGradientDescent()
48 const bool autotune = std::get<0>(GetParam()); in TEST_P() local
67 TF_ASSERT_OK(OptimizeWithEnableGradientDescent(item, &output, autotune)); in TEST_P()
73 (autotune && op != "_Retval") ? 1 : algorithm_index); in TEST_P()
Dmap_parallelization_test.cc32 GraphDef* output, bool autotune) { in OptimizeWithMapParallelization() argument
35 if (autotune) { in OptimizeWithMapParallelization()
51 const bool autotune = GetParam(); in TEST_P() local
70 TF_ASSERT_OK(OptimizeWithMapParallelization(item, &output, autotune)); in TEST_P()
72 autotune); in TEST_P()
73 EXPECT_EQ(graph_utils::ContainsGraphNodeWithName("map", output), !autotune); in TEST_P()
Dbatch_parallelization.h41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init() local
42 if (autotune == "true") { in Init()
44 } else if (autotune == "false") { in Init()
48 kAutotune, ": ", autotune); in Init()
Dautotune_buffer_sizes.h49 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init() local
50 if (autotune == "true") { in Init()
52 } else if (autotune == "false") { in Init()
56 kAutotune, ": ", autotune); in Init()
Ddisable_prefetch_legacy_autotune.h41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init() local
42 if (autotune == "true") { in Init()
44 } else if (autotune == "false") { in Init()
48 kAutotune, ": ", autotune); in Init()
Denable_gradient_descent.h41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init() local
42 if (autotune == "true") { in Init()
44 } else if (autotune == "false") { in Init()
48 kAutotune, ": ", autotune); in Init()
Dmap_parallelization.h41 const string& autotune = config->parameter_map().at(kAutotune).s(); in Init() local
42 if (autotune == "true") { in Init()
44 } else if (autotune == "false") { in Init()
48 kAutotune, ": ", autotune); in Init()
Dbatch_parallelization_test.cc32 GraphDef* output, bool autotune) { in OptimizeWithBatchParallelization() argument
35 if (autotune) { in OptimizeWithBatchParallelization()
49 const bool autotune = GetParam(); in TEST_P() local
69 TF_ASSERT_OK(OptimizeWithBatchParallelization(item, &output, autotune)); in TEST_P()
71 autotune); in TEST_P()
72 EXPECT_EQ(graph_utils::ContainsGraphNodeWithName("batch", output), !autotune); in TEST_P()
Dautotune_buffer_sizes_test.cc32 GraphDef *output, bool autotune) { in OptimizeWithAutotuneBufferSizes() argument
35 if (autotune) { in OptimizeWithAutotuneBufferSizes()
126 const bool autotune = GetParam(); in TEST_P() local
146 TF_ASSERT_OK(OptimizeWithAutotuneBufferSizes(item, &output, autotune)); in TEST_P()
148 autotune); in TEST_P()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doptimize_dataset_test.py259 options.experimental_optimization.autotune = apply_autotune
267 combinations.combine(autotune=False, autotune_buffers=False) +
268 combinations.combine(autotune=True, autotune_buffers=False) +
269 combinations.combine(autotune=True, autotune_buffers=True),
271 def testOptimizationEnableGradientDescent(self, autotune, autotune_buffers, argument
286 options.experimental_optimization.autotune = autotune
299 combinations.combine(autotune=[True, False, None]),
301 def testOptimizationMapParallelization(self, autotune, map_parallelization): argument
303 …if autotune is not False and map_parallelization is not False: # pylint: disable=g-bool-id-compar…
310 if autotune is not None:
[all …]
Dprefetch_with_slack_test.py50 dataset.options()._graph_rewrite_configs(autotune=True))
74 dataset.options()._graph_rewrite_configs(autotune=True))
Dmodel_dataset_test.py39 options.experimental_optimization.autotune = True
Dassert_next_test.py70 options.experimental_optimization.autotune = False
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py53 options.experimental_optimization.autotune = True
61 options1.experimental_optimization.autotune = True
67 self.assertTrue(ds.options().experimental_optimization.autotune)
77 options1.experimental_optimization.autotune = False
79 options2.experimental_optimization.autotune = True
83 self.assertTrue(ds.options().experimental_optimization.autotune)
88 options1.experimental_optimization.autotune = True
94 self.assertTrue(ds.options().experimental_optimization.autotune)
116 ds.options().experimental_optimization.autotune = True
117 self.assertTrue(ds.options().experimental_optimization.autotune)
[all …]
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto55 bool autotune = 2; field
57 // When autotuning is enabled (through autotune), determines whether to also
58 // autotune buffer sizes for datasets with parallelism.
62 // When autotuning is enabled (through autotune), determines the CPU budget to
68 // When autotuning is enabled (through autotune), determines the RAM budget to
Dmodel.h197 bool autotune() const TF_LOCKS_EXCLUDED(mu_) { in autotune() function
309 void set_autotune(bool autotune) TF_LOCKS_EXCLUDED(mu_) { in set_autotune() argument
310 autotune_.store(autotune); in set_autotune()
437 if (input->autotune()) { in num_inputs()
Dmodel_test.cc867 EXPECT_EQ(current->autotune(), cloned_current->autotune()); in TEST()
873 EXPECT_EQ(current->output()->autotune(), in TEST()
874 cloned_current->output()->autotune()); in TEST()
979 EXPECT_EQ(current->autotune(), restored_current->autotune()); in TEST()
996 EXPECT_EQ(current->output()->autotune(), in TEST()
997 restored_current->output()->autotune()); in TEST()
/external/tensorflow/tensorflow/python/data/benchmarks/
Drange_benchmark.py29 options.experimental_optimization.autotune = modeling_enabled
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-optimization-options.pbtxt11 name: "autotune"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-optimization-options.pbtxt11 name: "autotune"
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dlatency_all_edges_test.py45 options.experimental_optimization.autotune = False
/external/skqp/src/compute/hs/gen/
Dmain.c407 hsg_merge_levels_hint(struct hsg_merge * const merge, bool const autotune) in hsg_merge_levels_hint() argument
419 if (autotune) in hsg_merge_levels_hint()
1420 bool autotune = false; in main() local
1552 autotune = true; in main()
1615 hsg_merge_levels_hint(merge,autotune); in main()

12