Searched refs:stream_executor_ (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_internal.h | 87 SP_StreamExecutor stream_executor_; variable
98 stream_executor_(stream_executor), in CStream()
104 stream_executor_->create_stream(device_, &stream_handle_, c_status.get()); in Create()
111 stream_executor_->destroy_stream(device_, stream_handle_); in Destroy()
120 SP_StreamExecutor* stream_executor_; variable
128 stream_executor_(stream_executor), in CEvent()
134 stream_executor_->create_event(device_, &event_handle_, c_status.get()); in Create()
140 stream_executor_->record_event(device_, stream_handle, event_handle_, in Record()
147 stream_executor_->destroy_event(device_, event_handle_); in Destroy()
156 SP_StreamExecutor* stream_executor_; variable
[all …]
|
| D | stream_executor.cc | 246 stream_executor_(stream_executor), in CStreamExecutor()
263 stream_executor_->allocate(&device_, size, memory_space, &mem); in Allocate()
280 stream_executor_->deallocate(&device_, &device_memory_base); in Deallocate()
284 return stream_executor_->host_memory_allocate(&device_, size); in HostMemoryAllocate()
288 stream_executor_->host_memory_deallocate(&device_, mem); in HostMemoryDeallocate()
295 CHECK(stream_executor_->unified_memory_allocate); in UnifiedMemoryAllocate()
296 return stream_executor_->unified_memory_allocate(&device_, size); in UnifiedMemoryAllocate()
300 CHECK(stream_executor_->unified_memory_deallocate); in UnifiedMemoryDeallocate()
301 stream_executor_->unified_memory_deallocate(&device_, mem); in UnifiedMemoryDeallocate()
307 stream_executor_->get_allocator_stats(&device_, &c_stats); in GetAllocatorStats()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.h | 38 stream_executor_(stream_executor) {} in GpuLayoutAssignment()
54 se::StreamExecutor* stream_executor_; variable
|
| D | gpu_layout_assignment.cc | 156 HeuristicLayoutAssignment(instr, stream_executor_); in AddBackendConstraintsToDnnConvCustomCall()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 36 : stream_executor_(stream_executor) {} in XlaDeviceAllocator()
58 stream_executor_->GetAllocatorStats(); in GetStats()
|
| D | xla_device_context.h | 48 se::StreamExecutor* stream_executor_;
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.h | 137 se::StreamExecutor* stream_executor_; variable
|
| D | local_client_test_base.cc | 129 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform()) in LocalClientTestBase()
|
| D | transfer_manager_test.cc | 48 ->BorrowStream(stream_executor_)
|