1 // Copyright © 2022 Apple Inc. 2 3 #pragma once 4 #include <ATen/core/TensorBase.h> 5 6 namespace at::detail { 7 8 C10_EXPORT TensorBase empty_mps( 9 IntArrayRef size, 10 std::optional<ScalarType> dtype_opt, 11 std::optional<Layout> layout_opt, 12 std::optional<Device> device_opt, 13 std::optional<bool> pin_memory_opt, 14 std::optional<c10::MemoryFormat> memory_format_opt); 15 C10_EXPORT TensorBase empty_mps( 16 IntArrayRef size, const TensorOptions &options); 17 18 C10_EXPORT TensorBase empty_strided_mps( 19 IntArrayRef size, 20 IntArrayRef stride, 21 ScalarType dtype, 22 std::optional<Device> device_opt); 23 24 C10_EXPORT TensorBase empty_strided_mps( 25 IntArrayRef size, 26 IntArrayRef stride, 27 const TensorOptions &options); 28 29 } // namespace at::detail 30