Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_gpu_allocator.cc83 void* gpu_ptr = in Allocate() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc44 se::DeviceMemory<int64_t> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in CheckMask() local
66 se::DeviceMemory<int64_t> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in InitMask() local
Dgpu_util.cc429 DeviceMemoryBase gpu_ptr(ptr, num_bytes); in MemoryDebugString() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dstream_search_test.cc55 void* gpu_ptr = s.implementation()->GpuStreamHack(); in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddynamic_device_mgr_test.cc204 Device* gpu_ptr = gpu.get(); in TEST() local