| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 53 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()
|
| D | client_library.h | 63 LocalClientOptions& set_intra_op_parallelism_threads(int num_threads);
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 65 config.set_intra_op_parallelism_threads(1)
67 config.set_intra_op_parallelism_threads(2)
115 config.set_intra_op_parallelism_threads(1)
117 config.set_intra_op_parallelism_threads(2)
128 config.set_intra_op_parallelism_threads(10)
136 config.set_intra_op_parallelism_threads(1)
138 config.set_intra_op_parallelism_threads(10)
|
| D | config.py | 110 def set_intra_op_parallelism_threads(num_threads): function
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data_test.cc | 32 session_options.config.set_intra_op_parallelism_threads(2); in TEST()
|
| D | delegate.cc | 67 session_options.config.set_intra_op_parallelism_threads( in Initialize()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.threading.pbtxt | 16 name: "set_intra_op_parallelism_threads"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.threading.pbtxt | 16 name: "set_intra_op_parallelism_threads"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | basic_ops_benchmark_test.cc | 28 opts.config.set_intra_op_parallelism_threads(1); in InitOptions()
|
| 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()
|
| D | nn_ops_test.cc | 121 options.config.set_intra_op_parallelism_threads(num_threads); in BM_ConvFloat()
530 options.config.set_intra_op_parallelism_threads(num_threads); in BM_ConvFloatDepthwise()
1034 options.config.set_intra_op_parallelism_threads(num_threads); in BM_MaxPool()
1368 opts.config.set_intra_op_parallelism_threads(num_threads); in BM_ImageNetSoftmaxFwd()
1417 opts.config.set_intra_op_parallelism_threads(num_threads); in BM_TopK()
|
| D | reverse_op_test.cc | 180 opts.config.set_intra_op_parallelism_threads(intra_threads); in GetOptions()
|
| D | training_ops_test.cc | 29 opts->config.set_intra_op_parallelism_threads(num_threads); in InitMultiThreadingOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_util.cc | 110 config->set_intra_op_parallelism_threads(*num_threads); in CreateModuleConfig()
|
| D | backend.h | 55 BackendOptions& set_intra_op_parallelism_threads(int num_threads);
|
| D | hlo_module_config.h | 167 void set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function
|
| D | backend.cc | 49 BackendOptions& BackendOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads() function in xla::BackendOptions
|
| D | local_service.cc | 57 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
|
| D | service.h | 62 ServiceOptions& set_intra_op_parallelism_threads(int num_threads);
|
| D | hlo_runner.cc | 40 backend_options.set_intra_op_parallelism_threads( in HloRunner()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 61 client_options.set_intra_op_parallelism_threads( in BuildXlaCompilationCache()
|
| D | xla_device.cc | 248 .set_intra_op_parallelism_threads(intra_op_parallelism_threads_); in GetOrCreateClient()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 88 opts.config.set_intra_op_parallelism_threads(1); in BM_RunGraph()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 109 options.config.set_intra_op_parallelism_threads(1); in Cluster()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 53 options_.config.set_intra_op_parallelism_threads(num_cpu_cores); in SingleMachine()
|