Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py65 config.set_intra_op_parallelism_threads(1)
67 config.set_intra_op_parallelism_threads(2)
116 config.set_intra_op_parallelism_threads(1)
118 config.set_intra_op_parallelism_threads(2)
129 config.set_intra_op_parallelism_threads(10)
136 config.set_intra_op_parallelism_threads(1)
138 config.set_intra_op_parallelism_threads(10)
H A Dconfig.py108 def set_intra_op_parallelism_threads(num_threads): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient_library.cc55 LocalClientOptions& LocalClientOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::LocalClientOptions
111 service_options.set_intra_op_parallelism_threads( in GetOrCreateLocalClient()
H A Dclient_library.h61 LocalClientOptions& set_intra_op_parallelism_threads(int num_threads);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
H A Dbasic_ops_benchmark_test.cc28 opts.config.set_intra_op_parallelism_threads(1); in InitOptions()
H A Dnn_ops_test.cc125 options.config.set_intra_op_parallelism_threads(num_threads); in BM_ConvFloat()
536 options.config.set_intra_op_parallelism_threads(num_threads); in BM_ConvFloatDepthwise()
1049 options.config.set_intra_op_parallelism_threads(num_threads); in BM_MaxPool()
1397 opts.config.set_intra_op_parallelism_threads(num_threads); in BM_ImageNetSoftmaxFwd()
1454 opts.config.set_intra_op_parallelism_threads(num_threads); in BM_TopK()
H A Dreverse_op_test.cc180 opts.config.set_intra_op_parallelism_threads(intra_threads); in GetOptions()
H A Dtraining_ops_test.cc29 opts->config.set_intra_op_parallelism_threads(num_threads); in InitMultiThreadingOptions()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.config.threading.pbtxt16 name: "set_intra_op_parallelism_threads"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.config.threading.pbtxt16 name: "set_intra_op_parallelism_threads"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbackend.h53 BackendOptions& set_intra_op_parallelism_threads(int num_threads);
H A Dhlo_module_util.cc122 config->set_intra_op_parallelism_threads(*num_threads); in CreateModuleConfig()
H A Dbackend.cc48 BackendOptions& BackendOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::BackendOptions
H A Dlocal_service.cc56 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
H A Dhlo_module_config.h215 void set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function
H A Dservice.h61 ServiceOptions& set_intra_op_parallelism_threads(int num_threads);
H A Dhlo_runner.cc40 backend_options.set_intra_op_parallelism_threads( in HloRunner()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate.cc66 session_options.config.set_intra_op_parallelism_threads( in Initialize()
H A Ddelegate_data_test.cc49 session_options.config.set_intra_op_parallelism_threads(2); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device_test.cc88 opts.config.set_intra_op_parallelism_threads(1); in BM_RunGraph()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_platform_info.cc90 client_options.set_intra_op_parallelism_threads( in BuildXlaCompilationCache()
H A Dxla_device.cc257 .set_intra_op_parallelism_threads(intra_op_parallelism_threads_); in GetOrCreateClient()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Drpcbench_test.cc109 options.config.set_intra_op_parallelism_threads(1); in Cluster()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dsingle_machine.cc53 options_.config.set_intra_op_parallelism_threads(num_cpu_cores); in SingleMachine()

12