Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator.cc173 se::DeviceMemory<float> nan_ptr{ in AllocateRaw() local
177 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in AllocateRaw()
190 se::DeviceMemory<float> nan_ptr{ in DeallocateRaw() local
193 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in DeallocateRaw()