Home
last modified time | relevance | path

Searched refs:inter_op_parallelism_threads (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_util.cc75 int32_t inter_op_parallelism_threads = in InitComputePool() local
76 options.config.inter_op_parallelism_threads(); in InitComputePool()
77 if (inter_op_parallelism_threads == 0) { in InitComputePool()
78 inter_op_parallelism_threads = DefaultNumInterOpThreads(); in InitComputePool()
81 Env::Default(), ThreadOptions(), "Compute", inter_op_parallelism_threads, in InitComputePool()
126 const int32_t inter_op = options.config.inter_op_parallelism_threads(); in NumInterOpThreadsFromSessionOptions()
H A Ddirect_session.cc345 if (options_.config.inter_op_parallelism_threads() < 0 || in DirectSession()
346 (options_.config.inter_op_parallelism_threads() == 0 && in DirectSession()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig.py131 return context.context().inter_op_parallelism_threads
144 context.context().inter_op_parallelism_threads = num_threads
H A Dconfig_test.py144 context.context().inter_op_parallelism_threads)
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1078 config.inter_op_parallelism_threads = self._inter_op_parallelism_threads
1819 def inter_op_parallelism_threads(self): member in Context
1820 return self.config.inter_op_parallelism_threads
1822 @inter_op_parallelism_threads.setter
1823 def inter_op_parallelism_threads(self, num_threads): member in Context
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_coordinator_test.py126 session_config.inter_op_parallelism_threads += 1
825 session_config.inter_op_parallelism_threads)
835 session_config.inter_op_parallelism_threads)
H A Dmulti_worker_test_base.py136 if worker_config.inter_op_parallelism_threads < num_workers + 1:
137 worker_config.inter_op_parallelism_threads = num_workers + 1
H A Dstrategy_combinations.py260 worker_config.inter_op_parallelism_threads = 4 # max num_workers + 1
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
H A Dmap_and_batch_benchmark.py122 inter_op_parallelism_threads=inter_op, use_per_session_threads=True)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-config-proto.pbtxt19 name: "inter_op_parallelism_threads"
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto370 // inter_op_parallelism_threads is not configured), but still partitioning
455 // inter_op_parallelism_threads available in each process.
463 int32 inter_op_parallelism_threads = 5; field
473 // inter_op_parallelism_threads.
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
H A Dreader_ops_test.py713 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dimage_ops_test.py577 config.inter_op_parallelism_threads = 1
607 config.inter_op_parallelism_threads = 1
637 config.inter_op_parallelism_threads = 1
698 config.inter_op_parallelism_threads = 1
741 config.inter_op_parallelism_threads = 1
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Diterator_test.py179 inter_op_parallelism_threads=1, use_per_session_threads=True)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dwhile_v2_test.py1437 config.inter_op_parallelism_threads = -1
/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.h2648 ::int32_t inter_op_parallelism_threads() const; in Swap()
6309 inline ::int32_t ConfigProto::inter_op_parallelism_threads() const { in inter_op_parallelism_threads() function