1 #pragma once 2 3 #include <ATen/Tensor.h> 4 5 namespace torch::inductor { 6 7 TORCH_API at::Tensor _mm_plus_mm_out( 8 at::Tensor& out, 9 const at::Tensor& a, 10 const at::Tensor& b, 11 const at::Tensor& c, 12 const at::Tensor& d); 13 14 // After adding _mm_plus_mm_out, this should not be exposed and called by model 15 // code. Keeping it around for backward compatibility. Will be deprecated later. 16 TORCH_API at::Tensor _mm_plus_mm( 17 const at::Tensor& a, 18 const at::Tensor& b, 19 const at::Tensor& c, 20 const at::Tensor& d, 21 at::Tensor& out); 22 23 TORCH_API at::Tensor _alloc_from_pool( 24 const at::Tensor& self, 25 int64_t offset_bytes, 26 at::ScalarType dtype, 27 at::IntArrayRef size, 28 at::IntArrayRef stride); 29 30 // Similar to as_strided with the following differences 31 // - offset is added to the existing offset (rather than replacing it) 32 // - view tracking is disabled similar to unsafe_view 33 TORCH_API at::Tensor _reinterpret_tensor( 34 const at::Tensor& self, 35 at::IntArrayRef size, 36 at::IntArrayRef stride, 37 int64_t offset_increment = 0); 38 39 } // namespace torch::inductor 40