TensorConversions.h 652 B

12345678910111213141516171819202122232425
  1. #pragma once
  2. #include <c10/core/Device.h>
  3. #include <c10/core/Layout.h>
  4. #include <c10/core/MemoryFormat.h>
  5. #include <c10/core/ScalarType.h>
  6. #include <c10/util/Optional.h>
  7. namespace at {
  8. class Tensor;
  9. namespace native {
  10. bool to_will_alias(
  11. const Tensor& self,
  12. c10::optional<ScalarType> dtype,
  13. c10::optional<Layout> layout,
  14. c10::optional<Device> device,
  15. bool copy,
  16. c10::optional<c10::MemoryFormat> optional_memory_format);
  17. Tensor to_meta(const Tensor& tensor);
  18. c10::optional<Tensor> to_meta(const c10::optional<Tensor>& tensor);
  19. std::vector<Tensor> to_meta(at::ITensorListRef t_list);
  20. } // namespace native
  21. } // namespace at