Home
last modified time | relevance | path

Searched refs:set_intra_op_parallelism_threads (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py55 config.set_intra_op_parallelism_threads(1)
57 config.set_intra_op_parallelism_threads(2)
106 config.set_intra_op_parallelism_threads(1)
108 config.set_intra_op_parallelism_threads(2)
150 config.set_intra_op_parallelism_threads(10)
157 config.set_intra_op_parallelism_threads(1)
Dconfig.py86 def set_intra_op_parallelism_threads(num_threads): function
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient_library.cc53 LocalClientOptions& LocalClientOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::LocalClientOptions
109 service_options.set_intra_op_parallelism_threads( in GetOrCreateLocalClient()
Dclient_library.h63 LocalClientOptions& set_intra_op_parallelism_threads(int num_threads);
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data_test.cc32 session_options.config.set_intra_op_parallelism_threads(2); in TEST()
Ddelegate.cc37 session_options.config.set_intra_op_parallelism_threads( in Prepare()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.config.threading.pbtxt16 name: "set_intra_op_parallelism_threads"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.config.threading.pbtxt16 name: "set_intra_op_parallelism_threads"
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h97 void set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function
Dbackend.h55 BackendOptions& set_intra_op_parallelism_threads(int num_threads);
Dlocal_service.cc56 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
Dbackend.cc49 BackendOptions& BackendOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::BackendOptions
Dservice.h62 ServiceOptions& set_intra_op_parallelism_threads(int num_threads);
Dservice.cc113 ServiceOptions& ServiceOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::ServiceOptions
320 config->set_intra_op_parallelism_threads( in CreateModuleConfig()
/external/tensorflow/tensorflow/core/kernels/
Dbasic_ops_benchmark_test.cc28 opts.config.set_intra_op_parallelism_threads(1); in InitOptions()
Dsdca_ops_test.cc32 result->config.set_intra_op_parallelism_threads(1); in GetSingleThreadedOptions()
43 result->config.set_intra_op_parallelism_threads(0); // Auto-configured. in GetMultiThreadedOptions()
Dnn_ops_test.cc119 options.config.set_intra_op_parallelism_threads(num_threads); in BM_ConvFloat()
513 options.config.set_intra_op_parallelism_threads(num_threads); in BM_ConvFloatDepthwise()
1010 options.config.set_intra_op_parallelism_threads(num_threads); in BM_MaxPool()
1273 opts.config.set_intra_op_parallelism_threads(num_threads); in BM_ImageNetSoftmaxFwd()
1321 opts.config.set_intra_op_parallelism_threads(num_threads); in BM_TopK()
Dreverse_op_test.cc178 opts.config.set_intra_op_parallelism_threads(intra_threads); in GetOptions()
Dtraining_ops_test.cc28 opts.config.set_intra_op_parallelism_threads(1); in InitSingleThreadedOptions()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc87 opts.config.set_intra_op_parallelism_threads(1); in BM_RunGraph()
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc109 options.config.set_intra_op_parallelism_threads(1); in Cluster()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc53 options_.config.set_intra_op_parallelism_threads(num_cpu_cores); in SingleMachine()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc237 client_options.set_intra_op_parallelism_threads( in BuildCompilationCache()
/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops_test.cc491 opts.config.set_intra_op_parallelism_threads(1); in RunTensorFlowBenchmark()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc256 config.set_intra_op_parallelism_threads(num_threads); in InitializeSession()

12