/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | threadpool_dataset_op.cc | 46 int max_intra_op_parallelism) in ThreadPoolResource() argument 48 max_intra_op_parallelism_(max_intra_op_parallelism) {} in ThreadPoolResource() 271 int64_t max_intra_op_parallelism) in Dataset() argument 272 : Dataset(DatasetContext(ctx), input, max_intra_op_parallelism) {} in Dataset() 275 int64_t max_intra_op_parallelism) in Dataset() argument 278 max_intra_op_parallelism_(max_intra_op_parallelism), in Dataset() 385 OpKernelContext* ctx, DatasetBase* input, int32_t max_intra_op_parallelism, in MakeDatasetFromOptions() argument 388 ctx, max_intra_op_parallelism >= 0, in MakeDatasetFromOptions() 393 input, max_intra_op_parallelism); in MakeDatasetFromOptions() 399 int64_t max_intra_op_parallelism; in MakeDataset() local [all …]
|
D | threadpool_dataset_op.h | 35 int32_t max_intra_op_parallelism,
|
/external/tensorflow/tensorflow/core/data/ |
D | root_dataset.cc | 52 params.max_intra_op_parallelism = in FromOptions() 53 options.threading_options().max_intra_op_parallelism(); in FromOptions() 79 if (dataset()->params_.max_intra_op_parallelism >= 0) { in Iterator() 81 value_or_default(dataset()->params_.max_intra_op_parallelism, 0, in Iterator() 146 if (dataset()->params_.max_intra_op_parallelism >= 0) { in CreateParams() 202 if (params_.max_intra_op_parallelism >= 0) { 206 params_.max_intra_op_parallelism, 0,
|
D | root_dataset.h | 34 int64 max_intra_op_parallelism = 1; member
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | options_test.py | 161 options.threading.max_intra_op_parallelism = 30 195 opts.threading.max_intra_op_parallelism = 20 196 self.assertEqual(opts.experimental_threading.max_intra_op_parallelism, 20) 204 options.threading.max_intra_op_parallelism = 20 205 options.experimental_threading.max_intra_op_parallelism = 40 209 self.assertEqual(result.experimental_threading.max_intra_op_parallelism, 210 result.threading.max_intra_op_parallelism)
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | map_defun.py | 30 max_intra_op_parallelism=1): argument 68 max_intra_op_parallelism)
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | options.py | 371 max_intra_op_parallelism = options_lib.create_option( variable in ThreadingOptions 387 if self.max_intra_op_parallelism is not None: 388 pb.max_intra_op_parallelism = self.max_intra_op_parallelism 395 self.max_intra_op_parallelism = pb.max_intra_op_parallelism
|
/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"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
D | ExperimentalMaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
D | MaxIntraOpParallelismDataset.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"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | ExperimentalMaxIntraOpParallelismDataset.pbtxt | 8 name: "max_intra_op_parallelism"
|
D | MaxIntraOpParallelismDataset.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/v1/ |
D | tensorflow.data.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
D | tensorflow.data.experimental.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.data.-threading-options.pbtxt | 7 name: "max_intra_op_parallelism"
|
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()
|