Method Memset

tensorflow/stream_executor/host/host_gpu_executor.cc:130–138

}

port::Status HostExecutor::Memset(Stream *stream, DeviceMemoryBase *location,
                                  uint8 pattern, uint64 size) {
  void *gpu_mem = location->opaque();
  // Enqueue the [asynchronous] memzero on the stream (HostStream) associated
  // with the HostExecutor.
  AsHostStream(stream)->EnqueueTask(
      [gpu_mem, size, pattern]() { memset(gpu_mem, pattern, size); });
  return port::Status::OK();
}

port::Status HostExecutor::Memset32(Stream *stream, DeviceMemoryBase *location,
                                    uint32 pattern, uint64 size) {