Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
H A Dmap_parallelization_test.py66 options.experimental_optimization.map_parallelization = True
80 options.experimental_optimization.map_parallelization = True
94 options.experimental_optimization.map_parallelization = True
113 options.experimental_optimization.map_parallelization = True
131 options.experimental_optimization.map_parallelization = True
147 options.experimental_optimization.map_parallelization = True
H A Doptimization_test.py186 combinations.combine(map_parallelization=[True, False, None])))
187 def testOptimizationMapParallelization(self, autotune, map_parallelization): argument
189 …if autotune is not False and map_parallelization is not False: # pylint: disable=g-bool-id-compar…
198 if map_parallelization is not None:
199 options.experimental_optimization.map_parallelization = (
200 map_parallelization)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Doptions_test.py154 options.experimental_optimization.map_parallelization = True
274 combinations.combine(map_parallelization=[True, False])))
275 def testOptionsGraphRoundTripOptimization(self, map_parallelization): argument
278 options.experimental_optimization.map_parallelization = (
279 map_parallelization)
282 expected = "ParallelMap" if map_parallelization else "Map"
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Doptions.py365 map_parallelization = options_lib.create_option( variable in OptimizationOptions
406 if self.map_parallelization is not None:
407 pb.map_parallelization = self.map_parallelization
432 self.map_parallelization = pb.map_parallelization
H A Ddataset_ops.py479 options.experimental_optimization.map_parallelization = False
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A DBUILD33 ":map_parallelization",
707 name = "map_parallelization",
708 srcs = ["map_parallelization.cc"],
710 "map_parallelization.h",
736 ":map_parallelization",
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.data.experimental.-optimization-options.pbtxt35 name: "map_parallelization"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.data.experimental.-optimization-options.pbtxt35 name: "map_parallelization"
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset_options.proto118 bool map_parallelization = 12; field
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.cc150 if (optimization_options.map_parallelization()) { in DefaultOptimizationGraphRewrites()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Ddataset_options.pb.h968 bool map_parallelization() const; in Swap()
2173 inline bool OptimizationOptions::map_parallelization() const { in map_parallelization() function
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Dtest4.pbtxt2223 s: "map_parallelization:autotune:true"