Searched refs:default_stream_executor (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | reduction_vectorization_test.cc | 55 se::StreamExecutor* executor = backend().default_stream_executor(); in TEST_F()
103 se::StreamExecutor* executor = backend().default_stream_executor(); in TEST_F()
151 se::StreamExecutor* executor = backend().default_stream_executor(); in TEST_F()
235 se::StreamExecutor* executor = backend().default_stream_executor(); in TEST_F()
340 backend().default_stream_executor()->GetDeviceDescription(); in TEST_F()
|
| D | gpu_atomic_test.cc | 113 backend().default_stream_executor()->GetDeviceDescription(); in TEST_F()
|
| D | gpu_convolution_regression_test.cc | 31 backend().default_stream_executor(), backend().memory_allocator()); in CheckForHloText()
|
| D | gemm_rewrite_test.cc | 45 std::move(optimized_module), backend().default_stream_executor(), in CheckNumberOfAllocations()
46 backend().default_stream_executor()->GetAllocator())); in CheckNumberOfAllocations()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 81 backend().default_stream_executor()); in TEST_F()
170 backend().default_stream_executor()); in TEST_F()
241 backend().default_stream_executor()); in TEST_F()
323 backend().default_stream_executor()); in TEST_F()
360 backend().default_stream_executor()); in TEST_F()
398 backend().default_stream_executor()); in TEST_F()
425 backend().default_stream_executor()); in TEST_F()
|
| D | cudnn_fused_conv_rewriter_test.cc | 41 backend().default_stream_executor(), backend().memory_allocator()) in GetOptimizedHlo()
337 backend().default_stream_executor(), backend().memory_allocator()) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | codegen_test_base.cc | 24 backend().default_stream_executor(), in CompileToExecutable()
27 backend().default_stream_executor(), in CompileToExecutable()
|
| D | llvm_compiler_test.cc | 121 backend_->default_stream_executor(), in TestCompilerHooks()
143 executors.push_back({backend_->default_stream_executor()}); in TestMultiModuleCompilation()
144 executors.push_back({backend_->default_stream_executor()}); in TestMultiModuleCompilation()
|
| D | llvm_irgen_test_base.cc | 105 std::move(module), backend().default_stream_executor(), in GetOptimizedModule()
106 backend().default_stream_executor()->GetAllocator()); in GetOptimizedModule()
|
| D | buffer_donation_test.cc | 48 executor_ = backend_->default_stream_executor(); in BufferDonationTest()
|
| D | xla_hlo_profile_test.cc | 137 se::StreamExecutor* executor = backend->default_stream_executor(); in ExecuteAndFetchProfile()
|
| D | tuple_test.cc | 576 backend().default_stream_executor(), &literal)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 55 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralToDevice()
86 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralFromDevice()
163 se::Stream stream(backend().default_stream_executor()); in ExecuteWithDeviceBuffers()
433 {{backend().default_stream_executor()}}, in CreateExecutable()
438 backend().default_stream_executor(), in CreateExecutable()
|
| D | hlo_execution_profile_test.cc | 68 .default_stream_executor() in TEST_F()
|
| D | backend.h | 116 se::StreamExecutor* default_stream_executor() const { in default_stream_executor() function
|
| D | layout_assignment_test.cc | 681 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
688 backend().default_stream_executor(), in TEST_F()
937 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
969 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1002 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1035 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1059 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
|
| D | backend.cc | 148 return default_stream_executor()->device_ordinal(); in default_device_ordinal()
|
| D | service.cc | 822 execute_backend_->default_stream_executor(), in Compile()
869 execute_backend_->default_stream_executor())); in Execute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 130 std::move(hlo_module), backend->default_stream_executor(), in ConvertModule()
|