Home
last modified time | relevance | path

Searched refs:MASK_BYTES (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc27 #define MASK_BYTES (MASK_WORDS * sizeof(int64)) macro
44 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in CheckMask()
47 Status result = exec->SynchronousMemcpyD2H(gpu_ptr, MASK_BYTES, tmp); in CheckMask()
66 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in InitMask()
67 Status result = exec->SynchronousMemcpyH2D(mask, MASK_BYTES, &gpu_ptr); in InitMask()
88 num_bytes += (2 * MASK_BYTES); in AllocateRaw()
93 void* rv = static_cast<char*>(allocated_ptr) + MASK_BYTES; in AllocateRaw()
101 static_cast<char*>(allocated_ptr) + req_size - MASK_BYTES, in AllocateRaw()
111 ptr = static_cast<void*>(static_cast<char*>(ptr) - MASK_BYTES); in DeallocateRaw()
121 MASK_BYTES); in RequestedSize()
[all …]