Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.h57 static StatusOr<std::vector<se::StreamExecutor*>> GetStreamExecutors(
Dplatform_util.cc165 PlatformUtil::GetStreamExecutors( in GetStreamExecutors() function in xla::PlatformUtil
Dbackend.cc88 PlatformUtil::GetStreamExecutors(platform, options.allowed_devices())); in CreateBackend()
Dshaped_buffer_test.cc34 xla::PlatformUtil::GetStreamExecutors(platform)); in TEST()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.h50 platform, PlatformUtil::GetStreamExecutors(platform).ValueOrDie()) { in TestAllocator()
Dlocal_client_test_base.cc129 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform()) in LocalClientTestBase()
Ddynamic_ops_test.cc755 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_DynamicSlice()
Dlocal_client_execute_test.cc951 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_LocalClientOverhead()
Dcpu_gpu_fusion_test.cc831 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_ParallelFusion()
Dwhile_test.cc1266 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in BM_WhileLoop()
Ddot_operation_test.cc1815 auto executors = PlatformUtil::GetStreamExecutors(platform).ValueOrDie(); in DOT_ReorderContracting()