| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | override_threadpool_test.py | 76 num_threads=[1, 2, 4, 8, 16], max_intra_op_parallelism=[None]) +
78 num_threads=[4], max_intra_op_parallelism=[-1, 0, 4])))
79 def testNumThreadsDeprecated(self, num_threads, max_intra_op_parallelism): argument
86 max_intra_op_parallelism=max_intra_op_parallelism,
95 num_threads=[1, 2, 4, 8, 16], max_intra_op_parallelism=[None]) +
97 num_threads=[None], max_intra_op_parallelism=[0, 1, 4]) +
99 num_threads=[4], max_intra_op_parallelism=[0, 1, 4]) +
101 num_threads=[None], max_intra_op_parallelism=[None])))
102 def testNumThreads(self, num_threads, max_intra_op_parallelism): argument
106 if max_intra_op_parallelism is not None:
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | threading_options.py | 41 max_intra_op_parallelism = options.create_option( variable in ThreadingOptions
55 if self.max_intra_op_parallelism is not None:
56 pb.max_intra_op_parallelism = self.max_intra_op_parallelism
63 self.max_intra_op_parallelism = pb.max_intra_op_parallelism
|
| D | threadpool.py | 45 max_intra_op_parallelism=1): argument
51 max_intra_op_parallelism=max_intra_op_parallelism,
59 max_intra_op_parallelism=max_intra_op_parallelism,
|
| D | map_defun.py | 30 max_intra_op_parallelism=1): argument
68 max_intra_op_parallelism)
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 35 int max_intra_op_parallelism) in ThreadPoolResource() argument
37 max_intra_op_parallelism_(max_intra_op_parallelism) {} in ThreadPoolResource()
264 int64 max_intra_op_parallelism; in MakeDataset() local
267 &max_intra_op_parallelism)); in MakeDataset()
269 ctx, max_intra_op_parallelism >= 0, in MakeDataset()
271 *output = new Dataset(ctx, input, max_intra_op_parallelism); in MakeDataset()
278 int64 max_intra_op_parallelism) in Dataset() argument
281 max_intra_op_parallelism_(max_intra_op_parallelism) { in Dataset()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt | 4 name: "max_intra_op_parallelism"
|
| D | api_def_MaxIntraOpParallelismDataset.pbtxt | 4 name: "max_intra_op_parallelism"
|
| D | api_def_ThreadPoolHandle.pbtxt | 17 name: "max_intra_op_parallelism"
|
| D | api_def_ExperimentalThreadPoolHandle.pbtxt | 17 name: "max_intra_op_parallelism"
|
| D | api_def_MapDefun.pbtxt | 48 The `max_intra_op_parallelism` attr, which defaults to 1, can be used to
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
| D | ExperimentalMaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
| D | ThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| D | ExperimentalThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| D | MapDefun.pbtxt | 126 name: "max_intra_op_parallelism"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
| D | ExperimentalMaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
| D | ExperimentalThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| D | ThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| D | MapDefun.pbtxt | 126 name: "max_intra_op_parallelism"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 32 int max_intra_op_parallelism, string node_name) in MapDefunOpParams() argument
41 max_intra_op_parallelism_(max_intra_op_parallelism) {} in MapDefunOpParams()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | options_test.py | 161 options.experimental_threading.max_intra_op_parallelism = 30
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset_options.proto | 136 int32 max_intra_op_parallelism = 1; field
|