| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | threadpool_dataset_op.cc | 62 int max_intra_op_parallelism) in ThreadPoolResource() argument
64 max_intra_op_parallelism_(max_intra_op_parallelism) {} in ThreadPoolResource()
287 int64_t max_intra_op_parallelism) in Dataset() argument
288 : Dataset(DatasetContext(ctx), input, max_intra_op_parallelism) {} in Dataset()
291 int64_t max_intra_op_parallelism) in Dataset() argument
294 max_intra_op_parallelism_(max_intra_op_parallelism), in Dataset()
401 OpKernelContext* ctx, DatasetBase* input, int32_t max_intra_op_parallelism, in MakeDatasetFromOptions() argument
404 ctx, max_intra_op_parallelism >= 0, in MakeDatasetFromOptions()
409 input, max_intra_op_parallelism); in MakeDatasetFromOptions()
415 int64_t max_intra_op_parallelism; in MakeDataset() local
[all …]
|
| H A D | threadpool_dataset_op.h | 35 int32_t max_intra_op_parallelism,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | root_dataset.cc | 56 params->max_intra_op_parallelism = in SetRootDatasetParams()
57 options.threading_options().max_intra_op_parallelism(); in SetRootDatasetParams()
95 if (params.max_intra_op_parallelism >= 0) { in AddTraceMetadata()
99 params.max_intra_op_parallelism, 0, in AddTraceMetadata()
146 if (dataset()->params_.max_intra_op_parallelism >= 0) { in Iterator()
148 value_or_default(dataset()->params_.max_intra_op_parallelism, 0, in Iterator()
248 if (dataset()->params_.max_intra_op_parallelism >= 0) { in CreateParams()
|
| H A D | root_dataset.h | 35 int64_t max_intra_op_parallelism = 1; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | options_test.py | 159 options.threading.max_intra_op_parallelism = 30
193 opts.threading.max_intra_op_parallelism = 20
194 self.assertEqual(opts.experimental_threading.max_intra_op_parallelism, 20)
202 options.threading.max_intra_op_parallelism = 20
203 options.experimental_threading.max_intra_op_parallelism = 40
207 self.assertEqual(result.experimental_threading.max_intra_op_parallelism,
208 result.threading.max_intra_op_parallelism)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | options.py | 462 max_intra_op_parallelism = options_lib.create_option( variable in ThreadingOptions
478 if self.max_intra_op_parallelism is not None:
479 pb.max_intra_op_parallelism = self.max_intra_op_parallelism
486 self.max_intra_op_parallelism = pb.max_intra_op_parallelism
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| H A D | ExperimentalMaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
35 name: "max_intra_op_parallelism"
69 name: "max_intra_op_parallelism"
|
| H A D | MaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
35 name: "max_intra_op_parallelism"
69 name: "max_intra_op_parallelism"
|
| H A D | ExperimentalThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| H A D | ThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | map_defun.py | 26 max_intra_op_parallelism=1): argument
65 max_intra_op_parallelism)
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
| H A D | api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt | 4 name: "max_intra_op_parallelism"
|
| H A D | api_def_MaxIntraOpParallelismDataset.pbtxt | 4 name: "max_intra_op_parallelism"
|
| H A D | api_def_ExperimentalThreadPoolHandle.pbtxt | 17 name: "max_intra_op_parallelism"
|
| H A D | api_def_ThreadPoolHandle.pbtxt | 17 name: "max_intra_op_parallelism"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| H A D | ExperimentalMaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
| H A D | MaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
| H A D | ExperimentalThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| H A D | ThreadPoolHandle.pbtxt | 12 name: "max_intra_op_parallelism"
|
| H A D | MapDefun.pbtxt | 126 name: "max_intra_op_parallelism"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.data.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
| H A D | tensorflow.data.experimental.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.data.experimental.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
| H A D | tensorflow.data.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|