Viewing File: /home/ubuntu/combine_ai/combine/lib/python3.10/site-packages/torch/include/ATen/native/Copy.h

#pragma once

#include <ATen/native/DispatchStub.h>

namespace at {

class Tensor;
struct TensorIterator;
class TensorBase;

namespace native {

using copy_fn = void (*)(TensorIterator&, bool non_blocking);

DECLARE_DISPATCH(copy_fn, copy_stub);

TORCH_API void copy_ignoring_overlaps(const TensorBase &dst, const TensorBase &src);

} // namespace native
} // namespace at
Back to Directory File Manager