Home
last modified time | relevance | path

Searched refs:intra_op_parallelism_threads_ (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient_library.cc35 intra_op_parallelism_threads_(intra_op_parallelism_threads), in LocalClientOptions()
57 intra_op_parallelism_threads_ = num_threads; in set_intra_op_parallelism_threads()
62 return intra_op_parallelism_threads_; in intra_op_parallelism_threads()
H A Dclient_library.h73 int intra_op_parallelism_threads_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h217 intra_op_parallelism_threads_ = intra_op_parallelism_threads; in set_intra_op_parallelism_threads()
220 return intra_op_parallelism_threads_; in intra_op_parallelism_threads()
387 int64_t intra_op_parallelism_threads_ = -1; variable
H A Dbackend.cc50 intra_op_parallelism_threads_ = num_threads; in set_intra_op_parallelism_threads()
55 return intra_op_parallelism_threads_; in intra_op_parallelism_threads()
H A Dbackend.h64 int intra_op_parallelism_threads_ = -1; variable
H A Dservice.h73 int intra_op_parallelism_threads_ = -1; variable
H A Dservice.cc120 intra_op_parallelism_threads_ = num_threads; in set_intra_op_parallelism_threads()
125 return intra_op_parallelism_threads_; in intra_op_parallelism_threads()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc215 intra_op_parallelism_threads_( in XlaDevice()
257 .set_intra_op_parallelism_threads(intra_op_parallelism_threads_); in GetOrCreateClient()
H A Dxla_device.h233 const int intra_op_parallelism_threads_; variable
/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.cc4295 , decltype(_impl_.intra_op_parallelism_threads_){} in ConfigProto()
4323 ::memcpy(&_impl_.intra_op_parallelism_threads_, &from._impl_.intra_op_parallelism_threads_, in ConfigProto()
4325 …reinterpret_cast<char*>(&_impl_.intra_op_parallelism_threads_)) + sizeof(_impl_.share_cluster_devi… in ConfigProto()
4342 , decltype(_impl_.intra_op_parallelism_threads_){0} in SharedCtor()
4410 ::memset(&_impl_.intra_op_parallelism_threads_, 0, static_cast<size_t>( in Clear()
4412 …reinterpret_cast<char*>(&_impl_.intra_op_parallelism_threads_)) + sizeof(_impl_.share_cluster_devi… in Clear()
4439 … _impl_.intra_op_parallelism_threads_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint32(&ptr); in _InternalParse()
Dconfig.pb.h2729 ::int32_t intra_op_parallelism_threads_; in Swap()
6284 _impl_.intra_op_parallelism_threads_ = 0; in clear_intra_op_parallelism_threads()
6287 return _impl_.intra_op_parallelism_threads_; in _internal_intra_op_parallelism_threads()
6295 _impl_.intra_op_parallelism_threads_ = value; in _internal_set_intra_op_parallelism_threads()