EmptyTensor.h 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. #pragma once
  2. #include <ATen/core/TensorBase.h>
  3. namespace at {
  4. namespace detail {
  5. TORCH_CUDA_CPP_API TensorBase empty_cuda(
  6. IntArrayRef size,
  7. ScalarType dtype,
  8. c10::optional<Device> device_opt,
  9. c10::optional<c10::MemoryFormat> memory_format_opt);
  10. TORCH_CUDA_CPP_API TensorBase empty_cuda(
  11. IntArrayRef size,
  12. c10::optional<ScalarType> dtype_opt,
  13. c10::optional<Layout> layout_opt,
  14. c10::optional<Device> device_opt,
  15. c10::optional<bool> pin_memory_opt,
  16. c10::optional<c10::MemoryFormat> memory_format_opt);
  17. TORCH_CUDA_CPP_API TensorBase empty_cuda(
  18. IntArrayRef size,
  19. const TensorOptions &options);
  20. TORCH_CUDA_CPP_API TensorBase empty_strided_cuda(
  21. IntArrayRef size,
  22. IntArrayRef stride,
  23. ScalarType dtype,
  24. c10::optional<Device> device_opt);
  25. TORCH_CUDA_CPP_API TensorBase empty_strided_cuda(
  26. IntArrayRef size,
  27. IntArrayRef stride,
  28. c10::optional<ScalarType> dtype_opt,
  29. c10::optional<Layout> layout_opt,
  30. c10::optional<Device> device_opt,
  31. c10::optional<bool> pin_memory_opt);
  32. TORCH_CUDA_CPP_API TensorBase empty_strided_cuda(
  33. IntArrayRef size,
  34. IntArrayRef stride,
  35. const TensorOptions &options);
  36. }} // namespace at::detail