Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_parallelization_test.py70 options.experimental_optimization.map_parallelization = True
84 options.experimental_optimization.map_parallelization = True
98 options.experimental_optimization.map_parallelization = True
117 options.experimental_optimization.map_parallelization = True
135 options.experimental_optimization.map_parallelization = True
151 options.experimental_optimization.map_parallelization = True
Doptimization_test.py218 combinations.combine(map_parallelization=[True, False, None])))
219 def testOptimizationMapParallelization(self, autotune, map_parallelization): argument
221 …if autotune is not False and map_parallelization is not False: # pylint: disable=g-bool-id-compar…
230 if map_parallelization is not None:
231 options.experimental_optimization.map_parallelization = (
232 map_parallelization)
/external/tensorflow/tensorflow/python/data/ops/
Doptions.py277 map_parallelization = options_lib.create_option( variable in OptimizationOptions
319 if self.map_parallelization is not None:
320 pb.map_parallelization = self.map_parallelization
341 self.map_parallelization = pb.map_parallelization
Ddataset_ops.py397 options.experimental_optimization.map_parallelization = False
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py156 options.experimental_optimization.map_parallelization = True
276 combinations.combine(map_parallelization=[True, False])))
277 def testOptionsGraphRoundTripOptimization(self, map_parallelization): argument
280 options.experimental_optimization.map_parallelization = (
281 map_parallelization)
284 expected = "ParallelMap" if map_parallelization else "Map"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-optimization-options.pbtxt27 name: "map_parallelization"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-optimization-options.pbtxt27 name: "map_parallelization"
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
DBUILD30 ":map_parallelization",
564 name = "map_parallelization",
565 srcs = ["map_parallelization.cc"],
567 "map_parallelization.h",
593 ":map_parallelization",
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto90 bool map_parallelization = 12; field
/external/tensorflow/tensorflow/core/data/
Ddataset_utils.cc137 if (optimization_options.map_parallelization()) { in DefaultOptimizationGraphRewrites()