Searched refs:gpu_mem (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 102 void *gpu_mem = location->opaque(); in MemZero() local
106 [gpu_mem, size]() { memset(gpu_mem, 0, size); }); in MemZero()
112 void *gpu_mem = location->opaque(); in Memset() local
116 [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); }); in Memset()
122 void *gpu_mem = location->opaque(); in Memset32() local
126 [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); }); in Memset32()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_memory_test.cc | 123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F() local
124 EXPECT_EQ(16777216, gpu_mem.used_memory); in TEST_F()
126 for (const auto& t : gpu_mem.live_tensors) { in TEST_F()
149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F() local
151 EXPECT_EQ(20971520, gpu_mem.used_memory); in TEST_F()
153 for (const auto& t : gpu_mem.live_tensors) { in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 81 static hipDeviceptr_t AsROCmDevicePtr(const DeviceMemoryBase& gpu_mem) { in AsROCmDevicePtr() argument
82 return const_cast<hipDeviceptr_t>(gpu_mem.opaque()); in AsROCmDevicePtr()
86 static hipDeviceptr_t AsROCmDevicePtr(DeviceMemoryBase* gpu_mem) { in AsROCmDevicePtr() argument
87 return AsROCmDevicePtr(*gpu_mem); in AsROCmDevicePtr()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 104 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase &gpu_mem) { in AsCudaDevicePtr() argument
105 return reinterpret_cast<CUdeviceptr>(gpu_mem.opaque()); in AsCudaDevicePtr()
109 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase *gpu_mem) { in AsCudaDevicePtr() argument
110 return AsCudaDevicePtr(*gpu_mem); in AsCudaDevicePtr()
|