Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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()
H A Dexecutor_test.cc724 options.config.set_inter_op_parallelism_threads(4); in BM_WhileLoopHelper()
H A Ddirect_session_test.cc1486 options.config.set_inter_op_parallelism_threads(-1); in TEST()
2657 opts.config.set_inter_op_parallelism_threads(inter_op_threads); in FeedFetchBenchmarkHelper()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
H A Dvariable_ops_test.cc49 opts.config.set_inter_op_parallelism_threads(threads); in ManyManyVariablesHelper()
H A Dbasic_ops_benchmark_test.cc29 opts.config.set_inter_op_parallelism_threads(1); in InitOptions()
H A Dreverse_op_test.cc181 opts.config.set_inter_op_parallelism_threads(1); in GetOptions()
H A Dtraining_ops_test.cc30 opts->config.set_inter_op_parallelism_threads(1); in InitMultiThreadingOptions()
H A Dnn_ops_test.cc1396 opts.config.set_inter_op_parallelism_threads(1); in BM_ImageNetSoftmaxFwd()
1453 opts.config.set_inter_op_parallelism_threads(1); in BM_TopK()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.config.threading.pbtxt12 name: "set_inter_op_parallelism_threads"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.config.threading.pbtxt12 name: "set_inter_op_parallelism_threads"
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py142 config.set_inter_op_parallelism_threads(10)
149 config.set_inter_op_parallelism_threads(1)
151 config.set_inter_op_parallelism_threads(10)
H A Dconfig.py135 def set_inter_op_parallelism_threads(num_threads): function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate.cc64 session_options.config.set_inter_op_parallelism_threads(-1); in Initialize()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device_test.cc87 opts.config.set_inter_op_parallelism_threads(1); in BM_RunGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Drpcbench_test.cc110 options.config.set_inter_op_parallelism_threads(1); in Cluster()
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc262 config.set_inter_op_parallelism_threads(num_threads); in InitializeSession()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc164 config.set_inter_op_parallelism_threads(1); in TF_CreateConfig()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dconfig.pb.h2649 void set_inter_op_parallelism_threads(::int32_t value); in Swap()
6317 inline void ConfigProto::set_inter_op_parallelism_threads(::int32_t value) { in set_inter_op_parallelism_threads() function