Home
last modified time | relevance | path

Searched refs:GetStreamExecutors (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.h69 static StatusOr<std::vector<se::StreamExecutor*>> GetStreamExecutors(
Dshaped_buffer_test.cc36 xla::PlatformUtil::GetStreamExecutors(platform)); in TEST()
Dbackend.cc110 PlatformUtil::GetStreamExecutors(platform, options.allowed_devices())); in CreateBackend()
Dplatform_util.cc211 PlatformUtil::GetStreamExecutors( in GetStreamExecutors() function in xla::PlatformUtil
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.h46 platform, PlatformUtil::GetStreamExecutors(platform).ValueOrDie()) { in TestAllocator()
Dlocal_client_test_base.cc124 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform()) in LocalClientTestBase()
Ddynamic_ops_test.cc738 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_DynamicSlice()
Dlocal_client_execute_test.cc902 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_LocalClientOverhead()
Dfusion_test.cc832 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_ParallelFusion()
Dwhile_test.cc1267 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_WhileLoop()