Home
last modified time | relevance | path

Searched refs:map_parallelization (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_parallelization_test.py69 options.experimental_optimization.map_parallelization = True
83 options.experimental_optimization.map_parallelization = True
97 options.experimental_optimization.map_parallelization = True
116 options.experimental_optimization.map_parallelization = True
134 options.experimental_optimization.map_parallelization = True
150 options.experimental_optimization.map_parallelization = True
Dautotune_buffer_sizes_test.py107 options.experimental_optimization.map_parallelization = False
/external/tensorflow/tensorflow/python/data/experimental/ops/
Doptimization_options.py183 map_parallelization = options.create_option( variable in OptimizationOptions
373 if self.map_parallelization is not None:
374 pb.map_parallelization = self.map_parallelization
411 self.map_parallelization = pb.map_parallelization
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doptimize_dataset_test.py300 combinations.combine(map_parallelization=[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…
312 if map_parallelization is not None:
313 options.experimental_optimization.map_parallelization = (
314 map_parallelization)
462 options.experimental_optimization.map_parallelization = True
510 options.experimental_optimization.map_parallelization = False
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-optimization-options.pbtxt51 name: "map_parallelization"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-optimization-options.pbtxt51 name: "map_parallelization"
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
DBUILD31 ":map_parallelization",
662 name = "map_parallelization",
663 srcs = ["map_parallelization.cc"],
665 "map_parallelization.h",
690 ":map_parallelization",
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py152 options.experimental_optimization.map_parallelization = True
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto101 bool map_parallelization = 12; field