Home
last modified time | relevance | path

Searched refs:inter_op_parallelism_threads (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_util.cc49 int32 inter_op_parallelism_threads = in InitComputePool() local
50 options.config.inter_op_parallelism_threads(); in InitComputePool()
51 if (inter_op_parallelism_threads == 0) { in InitComputePool()
52 inter_op_parallelism_threads = DefaultNumInterOpThreads(); in InitComputePool()
55 inter_op_parallelism_threads); in InitComputePool()
78 const int32 inter_op = options.config.inter_op_parallelism_threads(); in NumInterOpThreadsFromSessionOptions()
/external/tensorflow/tensorflow/python/framework/
Dconfig.py109 return context.context().inter_op_parallelism_threads
122 context.context().inter_op_parallelism_threads = num_threads
Dconfig_test.py164 context.context().inter_op_parallelism_threads)
/external/tensorflow/tensorflow/python/eager/
Dcontext.py770 def inter_op_parallelism_threads(self): member in Context
771 return self._config.inter_op_parallelism_threads
773 @inter_op_parallelism_threads.setter
774 def inter_op_parallelism_threads(self, num_threads): member in Context
779 self._config.inter_op_parallelism_threads = num_threads
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.config.threading.pbtxt4 name: "inter_op_parallelism_threads"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.config.threading.pbtxt4 name: "inter_op_parallelism_threads"
Dtensorflow.-config-proto.pbtxt19 name: "inter_op_parallelism_threads"
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dmap_and_batch_benchmark.py151 inter_op_parallelism_threads=inter_op,
176 inter_op_parallelism_threads=inter_op,
/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/
Ddistort_image_ops_test.py262 config.inter_op_parallelism_threads = 1
306 config.inter_op_parallelism_threads = 1
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator_test.py38 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
325 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config.py312 inter_op_parallelism_threads=num_cores,
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_coordinator_test.py134 session_config.inter_op_parallelism_threads += 1
831 session_config.inter_op_parallelism_threads)
841 session_config.inter_op_parallelism_threads)
/external/tensorflow/tensorflow/contrib/eager/python/examples/l2hmc/
Dl2hmc_test.py167 session_conf = tf.ConfigProto(inter_op_parallelism_threads=1)
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto288 // inter_op_parallelism_threads is not configured), but still partitioning
328 // inter_op_parallelism_threads available in each process.
335 int32 inter_op_parallelism_threads = 5; field
345 // inter_op_parallelism_threads.
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dexperiment_test.py312 expected_config_proto.inter_op_parallelism_threads = 15
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_test.py489 config.inter_op_parallelism_threads = 1
519 config.inter_op_parallelism_threads = 1
549 config.inter_op_parallelism_threads = 1
610 config.inter_op_parallelism_threads = 1
653 config.inter_op_parallelism_threads = 1
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/
Dsdca_ops_test.py228 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_test.py178 inter_op_parallelism_threads=1, use_per_session_threads=True)
/external/tensorflow/tensorflow/python/kernel_tests/
Dreader_ops_test.py716 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_cell_test.py2058 inter_op_parallelism_threads=threads["inter"],
/external/tensorflow/tensorflow/python/keras/
Dbackend.py518 inter_op_parallelism_threads=num_thread,