Searched refs:backend_ (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 48 backend_(backend), in LocalExecutable()
89 if (stream_platform != backend_->platform()) { in ValidateExecutionOptions()
92 stream_platform->Name(), backend_->platform()->Name()); in ValidateExecutionOptions()
110 : backend_->default_device_ordinal(); in ValidateExecutionOptions()
113 backend_->devices_equivalent( in ValidateExecutionOptions()
117 backend_->stream_executor(run_device_ordinal)); in ValidateExecutionOptions()
119 backend_->stream_executor(build_device_ordinal())); in ValidateExecutionOptions()
123 backend_->device_name(build_device_ordinal()), in ValidateExecutionOptions()
125 backend_->device_name(run_device_ordinal), in ValidateExecutionOptions()
147 ValidateExecutionOptions(arguments, run_options, *backend_)); in Run()
[all …]
|
| D | local_client.h | 98 Backend* backend_ = nullptr; variable
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 42 backend_ = backend_or_status.ConsumeValueOrDie(); in SetUp()
79 backend_->default_stream_executor(), in TestCompilerHooks()
101 executors.push_back({backend_->default_stream_executor()}); in TestMultiModuleCompilation()
102 executors.push_back({backend_->default_stream_executor()}); in TestMultiModuleCompilation()
121 std::unique_ptr<Backend> backend_; member in xla::__anon16b34c660111::LLVMCompilerTest
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator_test.cc | 30 : backend_(Backend::CreateDefaultBackend().ConsumeValueOrDie()), in BufferComparatorTest()
31 stream_exec_(backend_->default_stream_executor()), in BufferComparatorTest()
71 std::unique_ptr<Backend> backend_; member in xla::gpu::__anon7e51ca1b0111::BufferComparatorTest
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.h | 44 AllocationTracker(Backend* backend) : backend_(backend), next_handle_(1) {} in AllocationTracker()
120 Backend* backend_; variable
|
| D | allocation_tracker.cc | 68 if (shaped_buffer.platform() != backend_->platform()) { in RegisterInternal()
72 backend_->platform()->Name(), shaped_buffer.platform()->Name()); in RegisterInternal()
225 backend_->memory_allocator()), in AddAllocationOrIncrementRefCount()
|
| D | hlo_runner.cc | 91 backend_ = Backend::CreateBackend(backend_options).ConsumeValueOrDie(); in HloRunner()
457 if (!backend_) { in backend()
458 backend_ = Backend::CreateDefaultBackend().ConsumeValueOrDie(); in backend()
461 return *backend_; in backend()
|
| D | execution_tracker.h | 54 Backend* backend_;
|
| D | execution_tracker.cc | 31 : backend_(CHECK_NOTNULL(backend)), in AsyncExecution()
|
| D | hlo_runner.h | 198 std::unique_ptr<Backend> backend_; variable
|