HomeSort by relevance Sort by last modified time
    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