Searched refs:nan_ptr (Results 1 – 1 of 1) sorted by relevance
173 se::DeviceMemory<float> nan_ptr{ in AllocateRaw() local177 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in AllocateRaw()190 se::DeviceMemory<float> nan_ptr{ in DeallocateRaw() local193 stream_exec_->SynchronousMemcpyH2D(&nans[0], req_size, &nan_ptr); in DeallocateRaw()