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

//  Copyright © 2022 Apple Inc.

#pragma once
#include <ATen/core/Tensor.h>

namespace at {
namespace native {
namespace mps {

at::Tensor& mps_copy_(at::Tensor& dst, const at::Tensor& src, bool non_blocking);
void copy_blit_mps(void* dst, const void* src, size_t size);

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