Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doverride_threadpool_test.py84 def testNumThreadsDeprecated(self, num_threads, max_intra_op_parallelism): argument
91 max_intra_op_parallelism=max_intra_op_parallelism,
110 def testNumThreads(self, num_threads, max_intra_op_parallelism): argument
114 if max_intra_op_parallelism is not None:
115 t_options.max_intra_op_parallelism = max_intra_op_parallelism
Dmap_defun_op_test.py54 max_intra_op_parallelism=0)[0]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dthreadpool.py45 max_intra_op_parallelism=1): argument
51 max_intra_op_parallelism=max_intra_op_parallelism,
59 max_intra_op_parallelism=max_intra_op_parallelism,
Dmap_defun.py30 max_intra_op_parallelism=1): argument
68 max_intra_op_parallelism)
Dthreading_options.py40 max_intra_op_parallelism = options.create_option( variable in ThreadingOptions
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc32 int max_intra_op_parallelism) in ThreadPoolResource() argument
34 max_intra_op_parallelism_(max_intra_op_parallelism) {} in ThreadPoolResource()
240 int64 max_intra_op_parallelism; in MakeDataset() local
243 &max_intra_op_parallelism)); in MakeDataset()
245 ctx, max_intra_op_parallelism >= 0, in MakeDataset()
247 *output = new Dataset(ctx, input, max_intra_op_parallelism); in MakeDataset()
254 int64 max_intra_op_parallelism) in Dataset() argument
257 max_intra_op_parallelism_(max_intra_op_parallelism) { in Dataset()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt4 name: "max_intra_op_parallelism"
Dapi_def_ExperimentalThreadPoolHandle.pbtxt17 name: "max_intra_op_parallelism"
Dapi_def_MapDefun.pbtxt48 The `max_intra_op_parallelism` attr, which defaults to 1, can be used to
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-threading-options.pbtxt7 name: "max_intra_op_parallelism"
Dtensorflow.raw_ops.pbtxt1121 …argspec: "args=[\'input_dataset\', \'max_intra_op_parallelism\', \'output_types\', \'output_shapes…
1189 …argspec: "args=[\'num_threads\', \'display_name\', \'max_intra_op_parallelism\', \'container\', \'…
1865 …aptured_inputs\', \'output_types\', \'output_shapes\', \'f\', \'max_intra_op_parallelism\', \'name…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-threading-options.pbtxt7 name: "max_intra_op_parallelism"
Dtensorflow.raw_ops.pbtxt1121 …argspec: "args=[\'input_dataset\', \'max_intra_op_parallelism\', \'output_types\', \'output_shapes…
1189 …argspec: "args=[\'num_threads\', \'display_name\', \'max_intra_op_parallelism\', \'container\', \'…
1865 …aptured_inputs\', \'output_types\', \'output_shapes\', \'f\', \'max_intra_op_parallelism\', \'name…
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py178 if t_options.max_intra_op_parallelism is not None:
180 dataset, t_options.max_intra_op_parallelism)
3332 def __init__(self, input_dataset, max_intra_op_parallelism): argument
3335 max_intra_op_parallelism,
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt11639 name: "max_intra_op_parallelism"
12244 name: "max_intra_op_parallelism"
17916 name: "max_intra_op_parallelism"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go38391 …ntraOpParallelismDataset(scope *Scope, input_dataset tf.Output, max_intra_op_parallelism tf.Output…
38399 input_dataset, max_intra_op_parallelism,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt24279 name: "max_intra_op_parallelism"
25048 name: "max_intra_op_parallelism"
34738 name: "max_intra_op_parallelism"
Dops_history.v2.pbtxt23908 name: "max_intra_op_parallelism"
24677 name: "max_intra_op_parallelism"