Home
last modified time | relevance | path

Searched defs:intra_op_parallelism_threads (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbackend.cc54 int BackendOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::BackendOptions
124 int intra_op_parallelism_threads) in Backend()
H A Dhlo_module_config.h216 const int intra_op_parallelism_threads) { in set_intra_op_parallelism_threads()
219 int64_t intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function
H A Dhlo_runner.cc37 HloRunner::HloRunner(se::Platform* platform, int intra_op_parallelism_threads) { in HloRunner()
H A Dservice.cc124 int ServiceOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::ServiceOptions
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient_library.cc31 int intra_op_parallelism_threads, in LocalClientOptions()
61 int LocalClientOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::LocalClientOptions
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlocal_device.cc76 int32_t intra_op_parallelism_threads = in EigenThreadPoolInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1804 def intra_op_parallelism_threads(self): member in Context
1808 def intra_op_parallelism_threads(self, num_threads): member in Context
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto452 int32 intra_op_parallelism_threads = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcpu_gpu_fusion_test.cc832 const int64_t intra_op_parallelism_threads = 24; in BM_ParallelFusion() local
/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.h6289 inline ::int32_t ConfigProto::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function