| /external/tensorflow/tensorflow/stream_executor/ |
| D | event.cc | 25 : stream_exec_(stream_exec), in Event()
27 stream_exec_->implementation()->CreateEventImplementation()) {} in Event()
31 if (stream_exec_ && implementation_) { in ~Event()
32 auto status = stream_exec_->DeallocateEvent(this); in ~Event()
40 auto status = stream_exec_->AllocateEvent(this); in Init()
50 return stream_exec_->PollForEventStatus(this); in PollForStatus()
|
| D | event.h | 72 StreamExecutor* stream_exec_; variable
|
| D | stream_executor_pimpl.cc | 78 : stream_exec_(stream_exec), in ScopedTracer()
81 if (stream_exec_->tracing_enabled_) { in ScopedTracer()
89 if (stream_exec_->tracing_enabled_) { in ~ScopedTracer()
99 absl::ReaderMutexLock lock{&stream_exec_->mu_}; in Trace()
100 for (TraceListener *listener : stream_exec_->listeners_) { in Trace()
107 StreamExecutor *stream_exec_; member in stream_executor::ScopedTracer
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_mem_allocator.h | 37 stream_exec_(stream_exec), in DeviceMemAllocator()
40 CHECK(stream_exec_ != nullptr); in DeviceMemAllocator()
50 ptr = stream_exec_->UnifiedMemoryAllocate(num_bytes); in Alloc()
52 ptr = stream_exec_->AllocateArray<char>(num_bytes).opaque(); in Alloc()
63 stream_exec_->UnifiedMemoryDeallocate(ptr); in Free()
66 stream_exec_->Deallocate(&device_ptr); in Free()
74 se::StreamExecutor* stream_exec_; // not owned, non-null
|
| D | device_host_allocator.h | 33 stream_exec_(stream_exec), in DeviceHostAllocator()
35 CHECK(stream_exec_ != nullptr); in DeviceHostAllocator()
44 ptr = stream_exec_->HostMemoryAllocate(num_bytes); in Alloc()
58 stream_exec_->HostMemoryDeallocate(ptr); in Free()
65 se::StreamExecutor* stream_exec_; // not owned, non-null
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.cc | 81 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUDebugAllocator()
97 InitMask(stream_exec_, allocated_ptr, before_mask); in AllocateRaw()
101 InitMask(stream_exec_, in AllocateRaw()
143 return CheckMask(stream_exec_, static_cast<char*>(ptr) - MASK_BYTES, in CheckHeader()
150 return CheckMask(stream_exec_, original_ptr + req_size - MASK_BYTES, in CheckFooter()
160 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUNanResetAllocator()
179 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in AllocateRaw()
195 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in DeallocateRaw()
|
| D | gpu_cudamalloc_allocator.cc | 32 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUcudaMallocAllocator()
42 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in AllocateRaw()
|
| D | gpu_debug_allocator.h | 56 se::StreamExecutor* stream_exec_; // Not owned. variable
80 se::StreamExecutor* stream_exec_; // Not owned. variable
|
| D | gpu_cudamalloc_allocator.h | 46 se::StreamExecutor* stream_exec_; // Not owned. variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 293 tensorflow::mutex_lock lock = LockGpu(stream_exec_); in PickBestAlgorithm()
299 ConvCacheKey key = AutotuneCacheKeyfromInstruction(instr, stream_exec_); in PickBestAlgorithm()
312 if (!stream_exec_->SynchronizeAllActivity()) { in PickBestAlgorithm()
323 se_allocator.emplace(stream_exec_); in PickBestAlgorithm()
328 allocator->GetStream(stream_exec_->device_ordinal())); in PickBestAlgorithm()
333 if (stream_exec_->platform_kind() == se::PlatformKind::kROCm) { in PickBestAlgorithm()
335 } else if (stream_exec_->platform_kind() == se::PlatformKind::kCuda) { in PickBestAlgorithm()
438 std::get<1>(AutotuneCacheKeyfromInstruction(instr, stream_exec_)); in PickBestAlgorithmNoCacheCuda()
441 if (auto* blas = stream_exec_->AsBlas()) { in PickBestAlgorithmNoCacheCuda()
446 GetComputeCapability(stream_exec_), GetCudnnVersion(stream_exec_), in PickBestAlgorithmNoCacheCuda()
[all …]
|
| D | gemm_algorithm_picker.h | 34 : stream_exec_(stream_exec), allocator_(allocator) {} in GemmAlgorithmPicker()
41 se::StreamExecutor* stream_exec_;
|
| D | buffer_comparator_test.cc | 36 stream_exec_(platform_->ExecutorForDevice(0).ValueOrDie()) {} in BufferComparatorTest()
42 se::Stream stream(stream_exec_); in CompareEqualBuffers()
46 stream_exec_->AllocateOwnedArray<ElementType>(lhs.size()); in CompareEqualBuffers()
48 stream_exec_->AllocateOwnedArray<ElementType>(rhs.size()); in CompareEqualBuffers()
79 se::StreamExecutor* stream_exec_; member in xla::gpu::__anon8d4987790111::BufferComparatorTest
|
| D | gpu_conv_algorithm_picker.h | 42 : stream_exec_(stream_exec), allocator_(allocator) {} in GpuConvAlgorithmPicker()
66 se::StreamExecutor* stream_exec_; // never null variable
|
| D | gemm_algorithm_picker.cc | 326 bool result, RunOnComputation(computation, stream_exec_, allocator_)); in Run()
|