Home
last modified time | relevance | path

Searched refs:max_intra_op_parallelism (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc46 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 …]
Dthreadpool_dataset_op.h35 int32_t max_intra_op_parallelism,
/external/tensorflow/tensorflow/core/data/
Droot_dataset.cc52 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,
Droot_dataset.h34 int64 max_intra_op_parallelism = 1; member
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py161 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/
Dmap_defun.py30 max_intra_op_parallelism=1): argument
68 max_intra_op_parallelism)
/external/tensorflow/tensorflow/python/data/ops/
Doptions.py371 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/
Dapi_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt4 name: "max_intra_op_parallelism"
Dapi_def_MaxIntraOpParallelismDataset.pbtxt4 name: "max_intra_op_parallelism"
Dapi_def_ThreadPoolHandle.pbtxt17 name: "max_intra_op_parallelism"
Dapi_def_ExperimentalThreadPoolHandle.pbtxt17 name: "max_intra_op_parallelism"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DExperimentalMaxIntraOpParallelismDataset.pbtxt8 name: "max_intra_op_parallelism"
DMaxIntraOpParallelismDataset.pbtxt8 name: "max_intra_op_parallelism"
DThreadPoolHandle.pbtxt12 name: "max_intra_op_parallelism"
DExperimentalThreadPoolHandle.pbtxt12 name: "max_intra_op_parallelism"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DExperimentalMaxIntraOpParallelismDataset.pbtxt8 name: "max_intra_op_parallelism"
DMaxIntraOpParallelismDataset.pbtxt8 name: "max_intra_op_parallelism"
DExperimentalThreadPoolHandle.pbtxt12 name: "max_intra_op_parallelism"
DThreadPoolHandle.pbtxt12 name: "max_intra_op_parallelism"
DMapDefun.pbtxt126 name: "max_intra_op_parallelism"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.-threading-options.pbtxt7 name: "max_intra_op_parallelism"
Dtensorflow.data.experimental.-threading-options.pbtxt7 name: "max_intra_op_parallelism"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.-threading-options.pbtxt7 name: "max_intra_op_parallelism"
Dtensorflow.data.experimental.-threading-options.pbtxt7 name: "max_intra_op_parallelism"
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op_test.cc32 int max_intra_op_parallelism, string node_name) in MapDefunOpParams() argument
41 max_intra_op_parallelism_(max_intra_op_parallelism) {} in MapDefunOpParams()

12