1 #pragma once 2 3 #include <ATen/core/Tensor.h> 4 5 namespace torch::utils { 6 7 TORCH_API void check_out_type_matches( 8 const at::Tensor& result, 9 std::optional<at::ScalarType> scalarType, 10 bool scalarType_is_none, 11 std::optional<at::Layout> layout, 12 std::optional<at::Device> device, 13 bool device_is_none); 14 15 } 16