1 #pragma once 2 3 #include <ATen/Tensor.h> 4 5 namespace torch::aot_inductor { 6 7 void* data_ptr_from_mkldnn(at::Tensor* mkldnn_tensor); 8 9 at::Tensor mkldnn_tensor_from_data_ptr( 10 void* data_ptr, 11 at::IntArrayRef dims, 12 at::ScalarType dtype, 13 at::Device device, 14 const uint8_t* opaque_metadata, 15 int64_t opaque_metadata_size); 16 17 } // namespace torch::aot_inductor 18