Program Listing for File algorithm.h

↰ Return to documentation for file (src/tensors/gpu/algorithm.h)

#pragma once

#include "tensors/backend.h"

namespace marian {
namespace gpu {
template <typename T>
void copy(Ptr<marian::Backend> backend, const T* begin, const T* end, T* dest);

template <typename T>
void fill(Ptr<marian::Backend> backend, T* begin, T* end, T value);

template <typename T>
void swap_ranges(Ptr<marian::Backend> backend, T* begin, T* end, T* dest);

void setSparse(Ptr<marian::Backend> backend,
               const std::vector<size_t>&,
               const std::vector<float>&,
               float*);
}  // namespace gpu
}  // namespace marian