Home
last modified time | relevance | path

Searched refs:autotune_buffers (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dautotune_benchmark.py31 def _run_benchmark(self, dataset, autotune, autotune_buffers, argument
36 options.experimental_optimization.autotune_buffers = autotune_buffers
41 if autotune_buffers else "parallelism_only")
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
69 autotune_buffers,
75 b = self._benchmark_map(autotune=True, autotune_buffers=False)
76 c = self._benchmark_map(autotune=True, autotune_buffers=True)
81 def _benchmark_map(self, autotune, autotune_buffers=False): argument
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doptimize_dataset_test.py267 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
287 options.experimental_optimization.autotune_buffers = autotune_buffers
322 combinations.combine(autotune=False, autotune_buffers=False) +
323 combinations.combine(autotune=True, autotune_buffers=False) +
324 combinations.combine(autotune=True, autotune_buffers=True),
332 def testOptimizationAutotuneBuffers(self, autotune, autotune_buffers, argument
342 options.experimental_optimization.autotune_buffers = autotune_buffers
[all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Doptimization_options.py118 autotune_buffers = options.create_option( variable in OptimizationOptions
233 if self.autotune_buffers is not None:
234 return self.autotune_buffers
311 autotune_buffers = self._autotune_buffers()
312 …if self.autotune is not False and autotune_buffers is True: # pylint: disable=g-bool-id-comparison
354 if self.autotune_buffers is not None:
355 pb.autotune_buffers = self.autotune_buffers
392 self.autotune_buffers = pb.autotune_buffers
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-optimization-options.pbtxt15 name: "autotune_buffers"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-optimization-options.pbtxt15 name: "autotune_buffers"
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dautotune_buffer_sizes_test.py34 options.experimental_optimization.autotune_buffers = True
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py143 options.experimental_optimization.autotune_buffers = True
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto60 bool autotune_buffers = 3; field