HomeSort by relevance Sort by last modified time
    Searched full:gpu_mem (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.cc 105 void *gpu_mem = location->opaque(); local
109 [gpu_mem, size]() { memset(gpu_mem, 0, size); });
115 void *gpu_mem = location->opaque(); local
119 [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); });
125 void *gpu_mem = location->opaque(); local
129 [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); });
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_memory_test.cc 123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); local
124 EXPECT_EQ(16777216, gpu_mem.used_memory);
126 for (const auto& t : gpu_mem.live_tensors) {
149 const GraphMemory::MemoryUsage& gpu_mem = local
151 EXPECT_EQ(20971520, gpu_mem.used_memory);
153 for (const auto& t : gpu_mem.live_tensors) {
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 102 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase &gpu_mem) {
103 return reinterpret_cast<CUdeviceptr>(gpu_mem.opaque());
107 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase *gpu_mem) {
108 return AsCudaDevicePtr(*gpu_mem);
    [all...]

Completed in 207 milliseconds