Home
last modified time | relevance | path

Searched refs:set_inter_op_parallelism_threads (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_util_test.cc24 opts.config.set_inter_op_parallelism_threads(10); in TEST()
30 opts.config.set_inter_op_parallelism_threads(10); in TEST()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.config.threading.pbtxt12 name: "set_inter_op_parallelism_threads"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.config.threading.pbtxt12 name: "set_inter_op_parallelism_threads"
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py161 config.set_inter_op_parallelism_threads(10)
168 config.set_inter_op_parallelism_threads(1)
Dconfig.py113 def set_inter_op_parallelism_threads(num_threads): function
/external/tensorflow/tensorflow/core/kernels/
Dvariable_ops_test.cc49 opts.config.set_inter_op_parallelism_threads(threads); in ManyManyVariablesHelper()
Dbasic_ops_benchmark_test.cc29 opts.config.set_inter_op_parallelism_threads(1); in InitOptions()
Dsdca_ops_test.cc33 result->config.set_inter_op_parallelism_threads(1); in GetSingleThreadedOptions()
44 result->config.set_inter_op_parallelism_threads(0); // Auto-configured. in GetMultiThreadedOptions()
Dreverse_op_test.cc179 opts.config.set_inter_op_parallelism_threads(1); in GetOptions()
Dtraining_ops_test.cc29 opts.config.set_inter_op_parallelism_threads(1); in InitSingleThreadedOptions()
Dnn_ops_test.cc1272 opts.config.set_inter_op_parallelism_threads(1); in BM_ImageNetSoftmaxFwd()
1320 opts.config.set_inter_op_parallelism_threads(1); in BM_TopK()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc86 opts.config.set_inter_op_parallelism_threads(1); in BM_RunGraph()
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc110 options.config.set_inter_op_parallelism_threads(1); in Cluster()
/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops_test.cc492 opts.config.set_inter_op_parallelism_threads(1); in RunTensorFlowBenchmark()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc257 config.set_inter_op_parallelism_threads(num_threads); in InitializeSession()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc104 config.set_inter_op_parallelism_threads(1); in TF_CreateConfig()