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