| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 38 backend_ = xla::Backend::CreateBackend(options).ConsumeValueOrDie(); in MlirGpuTestBase()
69 backend_->platform()->Name(), gpu_device_info, cuda_compute_capability, in CompileMlirModule()
76 static_cast<GpuCompiler*>(backend_->compiler()), module, "TestModule", in CompileMlirModule()
88 executable_run_options.set_allocator(backend_->memory_allocator()); in RunMlirModule()
112 auto* allocator = backend_->memory_allocator(); in RunMlirModuleWithHostBuffers()
118 ->Allocate(backend_->default_device_ordinal(), host_buffer.size()) in RunMlirModuleWithHostBuffers()
121 auto stream = backend_->BorrowStream(backend_->default_device_ordinal()) in RunMlirModuleWithHostBuffers()
168 auto stream = backend_->BorrowStream(backend_->default_device_ordinal()) in CompileMlirText()
|
| D | mlir_gpu_test_base.h | 62 std::unique_ptr<xla::Backend> backend_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 48 backend_(backend), in LocalExecutable()
64 if (stream_platform != backend_->platform()) { in ValidateExecutionOptions()
67 stream_platform->Name(), backend_->platform()->Name()); in ValidateExecutionOptions()
85 : backend_->default_device_ordinal(); in ValidateExecutionOptions()
88 backend_->devices_equivalent( in ValidateExecutionOptions()
92 backend_->stream_executor(run_device_ordinal)); in ValidateExecutionOptions()
94 backend_->stream_executor(build_device_ordinal())); in ValidateExecutionOptions()
98 backend_->device_name(build_device_ordinal()), in ValidateExecutionOptions()
100 backend_->device_name(run_device_ordinal), in ValidateExecutionOptions()
146 TF_RETURN_IF_ERROR(ValidateExecutionOptions(run_options, *backend_)); in RunHelper()
[all …]
|
| D | local_client.h | 121 Backend* backend_ = nullptr; variable
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 46 backend_ = &client_->backend(); in BufferDonationTest()
47 platform_ = backend_->platform(); in BufferDonationTest()
48 executor_ = backend_->default_stream_executor(); in BufferDonationTest()
55 const Backend* backend_; member in xla::__anon352f73640111::BufferDonationTest
69 TF_ASSERT_OK_AND_ASSIGN(hlo_module, backend_->compiler()->RunHloPasses( in RunAndCheck()
76 backend_->compiler()->RunBackend(std::move(hlo_module), executor_, in RunAndCheck()
83 platform_, backend_->stream_executors()); in RunAndCheck()
101 backend_->transfer_manager()->AllocateScopedShapedBuffer( in RunAndCheck()
105 TF_CHECK_OK(backend_->transfer_manager()->TransferLiteralToDevice( in RunAndCheck()
161 backend_->transfer_manager()->TransferLiteralFromDevice( in RunAndCheck()
|
| D | llvm_compiler_test.cc | 84 backend_ = backend_or_status.ConsumeValueOrDie(); in SetUp()
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()
158 std::unique_ptr<Backend> backend_; member in xla::__anon98f71c440111::LLVMCompilerTest
170 backend_->compiler()->ShapeSizeBytesFunction()); in CreateNewVerifiedModule()
|
| /external/drm_hwcomposer/backend/ |
| D | BackendManager.h | 27 #define REGISTER_BACKEND(name_str_, backend_) \ argument
32 return std::make_unique<backend_>(); \
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.h | 44 AllocationTracker(Backend* backend) : backend_(backend), next_handle_(1) {} in AllocationTracker()
121 Backend* backend_; variable
|
| D | execution_tracker.h | 54 Backend* backend_;
|
| D | hlo_runner.cc | 42 backend_ = Backend::CreateBackend(backend_options).ConsumeValueOrDie(); in HloRunner()
459 if (!backend_) { in backend()
460 backend_ = Backend::CreateDefaultBackend().ConsumeValueOrDie(); in backend()
463 return *backend_; in backend()
|
| D | execution_tracker.cc | 31 : backend_(CHECK_NOTNULL(backend)), in AsyncExecution()
|
| D | hlo_runner.h | 163 std::unique_ptr<Backend> backend_; variable
|
| D | allocation_tracker.cc | 213 backend_->memory_allocator()), in AddAllocationOrIncrementRefCount()
|
| /external/minigbm/ |
| D | dumb_driver.c | 12 const struct backend backend_##driver = { \
|
| /external/drm_hwcomposer/ |
| D | DrmHwcTwo.h | 266 return backend_.get(); in backend()
269 backend_ = std::move(backend); in set_backend()
326 std::unique_ptr<Backend> backend_; variable
|
| D | DrmHwcTwo.cpp | 892 return backend_->ValidateDisplay(this, num_types, num_requests); in ValidateDisplay()
|