Home
last modified time | relevance | path

Searched refs:eigen_intra_op_thread_pool_device (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbackend.h153 const Eigen::ThreadPoolDevice* eigen_intra_op_thread_pool_device() const;
H A Dbackend.cc150 const Eigen::ThreadPoolDevice* Backend::eigen_intra_op_thread_pool_device() in eigen_intra_op_thread_pool_device() function in xla::Backend
H A Dservice.cc432 backend->eigen_intra_op_thread_pool_device()); in ExecuteParallelAndRegisterResult()
530 backend->eigen_intra_op_thread_pool_device()); in ExecuteAndRegisterResult()
H A Dhlo_runner.cc480 backend().eigen_intra_op_thread_pool_device()); in GetServiceRunOptionsForDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dmultiple_devices_on_host_test.cc42 client->backend().eigen_intra_op_thread_pool_device()); in CompileAndExecute()
H A Dxla_hlo_profile_test.cc174 backend->eigen_intra_op_thread_pool_device()); in ExecuteAndFetchProfile()
H A Dexhaustive_op_test_utils.h278 client_->backend().eigen_intra_op_thread_pool_device());
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc1980 client_->client()->backend().eigen_intra_op_thread_pool_device()); in EnqueueExecution()