| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 80 &computation_layout, backend().default_stream_executor()); in TEST_F()
168 &computation_layout, backend().default_stream_executor()); in TEST_F()
238 &computation_layout, backend().default_stream_executor()); in TEST_F()
319 &computation_layout, backend().default_stream_executor()); in TEST_F()
355 backend().default_stream_executor()); in TEST_F()
392 backend().default_stream_executor()); in TEST_F()
418 backend().default_stream_executor()); in TEST_F()
|
| D | cudnn_fused_conv_rewriter_test.cc | 53 backend().default_stream_executor(), in GetOptimizedHlo()
354 backend().default_stream_executor(), backend().memory_allocator()) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_too_many_blocks_test.cc | 50 std::move(optimized_module), backend().default_stream_executor(), in TEST_F()
51 backend().default_stream_executor()->GetAllocator()); in TEST_F()
|
| D | gemm_rewrite_test.cc | 46 std::move(optimized_module), backend().default_stream_executor(), in CheckNumberOfAllocations()
47 backend().default_stream_executor()->GetAllocator())); in CheckNumberOfAllocations()
57 .default_stream_executor() in GetCudaComputeCapability()
127 *get_module(), backend().default_stream_executor(),
128 backend().default_stream_executor()->GetAllocator()));
|
| D | gpu_atomic_test.cc | 114 .default_stream_executor() in TEST_F()
|
| D | reduction_vectorization_test.cc | 82 .default_stream_executor() in TEST_F()
|
| D | gpu_convolution_regression_test.cc | 31 backend().default_stream_executor(), backend().memory_allocator()); in CheckForHloText()
|
| /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 | 597 backend().default_stream_executor(), &literal)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 59 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralToDevice()
90 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralFromDevice()
169 se::Stream stream(backend().default_stream_executor()); in ExecuteWithDeviceBuffers()
449 {{backend().default_stream_executor()}}, in CreateExecutable()
454 backend().default_stream_executor(), in CreateExecutable()
|
| D | hlo_execution_profile_test.cc | 68 .default_stream_executor() in TEST_F()
|
| 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.h | 119 se::StreamExecutor* default_stream_executor() const { in default_stream_executor() function
|
| D | backend.cc | 148 return default_stream_executor()->device_ordinal(); in default_device_ordinal()
|
| D | service.cc | 800 execute_backend_->default_stream_executor(), in Compile()
847 execute_backend_->default_stream_executor())); in Execute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 138 std::move(hlo_module), backend->default_stream_executor(), in OptimizeAndConvertHloToLmhlo()
|