Home
last modified time | relevance | path

Searched defs:gpu_ptr (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc44 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in CheckMask() local
66 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in InitMask() local
Dgpu_util.cc383 DeviceMemoryBase gpu_ptr(ptr, num_bytes); in MemoryDebugString() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_gpu_allocator.cc83 void* gpu_ptr = in Allocate() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddynamic_device_mgr_test.cc204 Device* gpu_ptr = gpu.get(); in TEST() local