TensorConversions.h 830 B

1234567891011121314151617181920212223242526
  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. std::optional<ScalarType> dtype,
  13. std::optional<Layout> layout,
  14. std::optional<Device> device,
  15. bool copy,
  16. std::optional<c10::MemoryFormat> optional_memory_format);
  17. Tensor to_meta(const Tensor& tensor);
  18. std::optional<Tensor> to_meta(const std::optional<Tensor>& tensor);
  19. std::vector<Tensor> to_meta(at::ITensorListRef t_list);
  20. Tensor dense_to_sparse_with_mask(const Tensor& self, const Tensor& mask, std::optional<c10::Layout> layout, OptionalIntArrayRef blocksize, std::optional<int64_t> dense_dim_opt);
  21. } // namespace native
  22. } // namespace at