OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:UnifiedMemoryAllocate
(Results
1 - 8
of
8
) sorted by null
/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_bfc_allocator.h
56
ptr = stream_exec_->
UnifiedMemoryAllocate
(num_bytes);
/external/tensorflow/tensorflow/stream_executor/gpu/
gpu_executor.h
96
void*
UnifiedMemoryAllocate
(uint64 size) override {
97
return GpuDriver::
UnifiedMemoryAllocate
(context_, size);
gpu_driver.h
115
static void*
UnifiedMemoryAllocate
(GpuContext* context, uint64 bytes);
/external/tensorflow/tensorflow/stream_executor/
stream_executor_internal.h
206
virtual void *
UnifiedMemoryAllocate
(uint64 size) { return nullptr; }
209
//
UnifiedMemoryAllocate
.
stream_executor_pimpl.cc
548
void *StreamExecutor::
UnifiedMemoryAllocate
(uint64 bytes) {
549
void *buffer = implementation_->
UnifiedMemoryAllocate
(bytes);
550
VLOG(1) << "Called StreamExecutor::
UnifiedMemoryAllocate
(size=" << bytes
stream_executor_pimpl.h
216
void *
UnifiedMemoryAllocate
(uint64 bytes);
219
//
UnifiedMemoryAllocate
.
[
all
...]
/external/tensorflow/tensorflow/stream_executor/rocm/
rocm_driver.cc
684
/* static */ void* GpuDriver::
UnifiedMemoryAllocate
(GpuContext* context,
689
<< "Feature not supported on ROCm platform (
UnifiedMemoryAllocate
)";
[
all
...]
/external/tensorflow/tensorflow/stream_executor/cuda/
cuda_driver.cc
860
/* static */ void* GpuDriver::
UnifiedMemoryAllocate
(GpuContext* context,
[
all
...]
Completed in 350 milliseconds