/aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
H A D | config_test.py | 65 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 D | config.py | 108 def set_intra_op_parallelism_threads(num_threads): function
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | client_library.cc | 55 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 D | client_library.h | 61 LocalClientOptions& set_intra_op_parallelism_threads(int num_threads);
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | sdca_ops_test.cc | 32 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 D | basic_ops_benchmark_test.cc | 28 opts.config.set_intra_op_parallelism_threads(1); in InitOptions()
|
H A D | nn_ops_test.cc | 125 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 D | reverse_op_test.cc | 180 opts.config.set_intra_op_parallelism_threads(intra_threads); in GetOptions()
|
H A D | training_ops_test.cc | 29 opts->config.set_intra_op_parallelism_threads(num_threads); in InitMultiThreadingOptions()
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.config.threading.pbtxt | 16 name: "set_intra_op_parallelism_threads"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.config.threading.pbtxt | 16 name: "set_intra_op_parallelism_threads"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | backend.h | 53 BackendOptions& set_intra_op_parallelism_threads(int num_threads);
|
H A D | hlo_module_util.cc | 122 config->set_intra_op_parallelism_threads(*num_threads); in CreateModuleConfig()
|
H A D | backend.cc | 48 BackendOptions& BackendOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::BackendOptions
|
H A D | local_service.cc | 56 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
|
H A D | hlo_module_config.h | 215 void set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function
|
H A D | service.h | 61 ServiceOptions& set_intra_op_parallelism_threads(int num_threads);
|
H A D | hlo_runner.cc | 40 backend_options.set_intra_op_parallelism_threads( in HloRunner()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
H A D | delegate.cc | 66 session_options.config.set_intra_op_parallelism_threads( in Initialize()
|
H A D | delegate_data_test.cc | 49 session_options.config.set_intra_op_parallelism_threads(2); in TEST()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
H A D | kernel_and_device_test.cc | 88 opts.config.set_intra_op_parallelism_threads(1); in BM_RunGraph()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
H A D | xla_platform_info.cc | 90 client_options.set_intra_op_parallelism_threads( in BuildXlaCompilationCache()
|
H A D | xla_device.cc | 257 .set_intra_op_parallelism_threads(intra_op_parallelism_threads_); in GetOrCreateClient()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
H A D | rpcbench_test.cc | 109 options.config.set_intra_op_parallelism_threads(1); in Cluster()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
H A D | single_machine.cc | 53 options_.config.set_intra_op_parallelism_threads(num_cpu_cores); in SingleMachine()
|